cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 530751097
This commit is contained in:
MediaPipe Team 2023-05-09 16:59:44 -07:00 committed by Copybara-Service
parent dd1779840d
commit 6ff39c418c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/tasks/java/com/google/mediapipe/tasks/core/TaskOptions.java
View File

@ -64,7 +64,9 @@ public abstract class TaskOptions {
break; break;
case GPU: case GPU:
accelerationBuilder.setGpu( accelerationBuilder.setGpu(
InferenceCalculatorProto.InferenceCalculatorOptions.Delegate.Gpu.getDefaultInstance()); InferenceCalculatorProto.InferenceCalculatorOptions.Delegate.Gpu.newBuilder()
.setUseAdvancedGpuApi(true)
.build());
break; break;
} }
return BaseOptionsProto.BaseOptions.newBuilder() return BaseOptionsProto.BaseOptions.newBuilder()