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

Internal change

Browse Source 
PiperOrigin-RevId: 530408554
This commit is contained in:
MediaPipe Team 2023-05-08 14:28:53 -07:00 committed by Copybara-Service
parent b40f0d3b72
commit 83a8743a8b
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/cc/core/base_options.cc
View File

@ -55,7 +55,9 @@ proto::BaseOptions ConvertBaseOptionsToProto(BaseOptions* base_options) {
base_options_proto.mutable_acceleration()->mutable_tflite(); base_options_proto.mutable_acceleration()->mutable_tflite();
break; break;
case BaseOptions::Delegate::GPU: case BaseOptions::Delegate::GPU:
base_options_proto.mutable_acceleration()->mutable_gpu(); base_options_proto.mutable_acceleration()
->mutable_gpu()
->set_use_advanced_gpu_api(true);
break; break;
case BaseOptions::Delegate::EDGETPU_NNAPI: case BaseOptions::Delegate::EDGETPU_NNAPI:
base_options_proto.mutable_acceleration() base_options_proto.mutable_acceleration()