Internal change
PiperOrigin-RevId: 530751097
This commit is contained in:
parent
dd1779840d
commit
6ff39c418c
|
@ -64,7 +64,9 @@ public abstract class TaskOptions {
|
|||
break;
|
||||
case GPU:
|
||||
accelerationBuilder.setGpu(
|
||||
InferenceCalculatorProto.InferenceCalculatorOptions.Delegate.Gpu.getDefaultInstance());
|
||||
InferenceCalculatorProto.InferenceCalculatorOptions.Delegate.Gpu.newBuilder()
|
||||
.setUseAdvancedGpuApi(true)
|
||||
.build());
|
||||
break;
|
||||
}
|
||||
return BaseOptionsProto.BaseOptions.newBuilder()
|
||||
|
|
Loading…
Reference in New Issue
Block a user