Home
last modified time | relevance | path

Searched refs:delegate_options (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/
Ddelegate.cc294 auto delegate_options = delegate_->options(); in InitializeOpenClApi() local
298 if (delegate_options.is_precision_loss_allowed == -1) { in InitializeOpenClApi()
299 options.priority1 = ToPriority(delegate_options.inference_priority1); in InitializeOpenClApi()
300 options.priority2 = ToPriority(delegate_options.inference_priority2); in InitializeOpenClApi()
301 options.priority3 = ToPriority(delegate_options.inference_priority3); in InitializeOpenClApi()
304 if (delegate_options.is_precision_loss_allowed == 0) { in InitializeOpenClApi()
310 options.usage = ToUsage(delegate_options.inference_preference); in InitializeOpenClApi()
326 auto delegate_options = delegate_->options(); in InitializeOpenGlApi() local
328 options.usage = ToUsage(delegate_options.inference_preference); in InitializeOpenGlApi()
329 options.priority1 = ToPriority(delegate_options.inference_priority1); in InitializeOpenGlApi()
[all …]
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate.cc82 auto delegate_options = delegate->DelegateOptions(); in DelegatePrepare() local
83 if (delegate_options.max_delegated_partitions <= 0) in DelegatePrepare()
84 delegate_options.max_delegated_partitions = std::numeric_limits<int>::max(); in DelegatePrepare()
98 delegate_options.max_delegated_partitions, in DelegatePrepare()
99 delegate_options.min_nodes_per_partition); in DelegatePrepare()
/external/tensorflow/tensorflow/lite/tools/delegates/
Dexternal_delegate_provider.cc84 auto delegate_options = in CreateTfLiteDelegate() local
106 delegate_options.insert(&delegate_options, keys.back().c_str(), in CreateTfLiteDelegate()
111 auto external_delegate = TfLiteExternalDelegateCreate(&delegate_options); in CreateTfLiteDelegate()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dresize_bilinear_test.cc95 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
97 delegate_options.num_threads = 2; in TEST()
99 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dlogistic_test.cc101 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
103 delegate_options.num_threads = 2; in TEST()
105 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Drelu6_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Drelu_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dsquare_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dneg_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Drelu_n1_to_1_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dfloor_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dceil_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dsqrt_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Delu_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dabs_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dhard_swish_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dround_test.cc98 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
100 delegate_options.num_threads = 2; in TEST()
102 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dleaky_relu_test.cc115 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
117 delegate_options.num_threads = 2; in TEST()
119 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dsoftmax_test.cc118 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
120 delegate_options.num_threads = 2; in TEST()
122 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Ddepth_to_space_test.cc128 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
130 delegate_options.num_threads = 2; in TEST()
132 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dreshape_test.cc201 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
203 delegate_options.num_threads = 2; in TEST()
205 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dpad_test.cc257 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
259 delegate_options.num_threads = 2; in TEST()
261 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dfully_connected_test.cc324 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
326 delegate_options.num_threads = 2; in TEST()
328 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
Dmax_pool_2d_test.cc391 TfLiteXNNPackDelegateOptions delegate_options = in TEST() local
393 delegate_options.num_threads = 2; in TEST()
395 xnnpack_delegate(TfLiteXNNPackDelegateCreate(&delegate_options), in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc583 bool ShouldUseTargetDevices(StatefulNnApiDelegate::Options delegate_options, in ShouldUseTargetDevices() argument
586 const char* device_name_ptr = delegate_options.accelerator_name; in ShouldUseTargetDevices()
592 return (delegate_options.disallow_nnapi_cpu && in ShouldUseTargetDevices()
616 const auto delegate_options = StatefulNnApiDelegate::GetOptions(delegate); in GetTargetDevices() local
617 const char* device_name_ptr = delegate_options.accelerator_name; in GetTargetDevices()
625 } else if (delegate_options.disallow_nnapi_cpu) { in GetTargetDevices()
3623 const auto delegate_options = in Init() local
3626 ShouldUseTargetDevices(delegate_options, nnapi_)) { in Init()
3648 TF_LITE_ENSURE_STATUS(BuildGraph(context, delegate_options, in Init()
3656 const char* cache_dir = delegate_options.cache_dir; in Init()
[all …]

12