Internal change
PiperOrigin-RevId: 530408554
This commit is contained in:
parent
b40f0d3b72
commit
83a8743a8b
|
@ -55,7 +55,9 @@ proto::BaseOptions ConvertBaseOptionsToProto(BaseOptions* base_options) {
|
|||
base_options_proto.mutable_acceleration()->mutable_tflite();
|
||||
break;
|
||||
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;
|
||||
case BaseOptions::Delegate::EDGETPU_NNAPI:
|
||||
base_options_proto.mutable_acceleration()
|
||||
|
|
Loading…
Reference in New Issue
Block a user