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();
|
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()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user