Home
last modified time | relevance | path

Searched refs:accelerator_name (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/java/src/main/native/
Dnnapi_delegate_jni.cc26 JNIEnv* env, jclass clazz, jint preference, jstring accelerator_name, in Java_org_tensorflow_lite_nnapi_NnApiDelegate_createDelegate() argument
33 if (accelerator_name) { in Java_org_tensorflow_lite_nnapi_NnApiDelegate_createDelegate()
34 options.accelerator_name = env->GetStringUTFChars(accelerator_name, NULL); in Java_org_tensorflow_lite_nnapi_NnApiDelegate_createDelegate()
57 if (options.accelerator_name) { in Java_org_tensorflow_lite_nnapi_NnApiDelegate_createDelegate()
58 env->ReleaseStringUTFChars(accelerator_name, options.accelerator_name); in Java_org_tensorflow_lite_nnapi_NnApiDelegate_createDelegate()
62 env->ReleaseStringUTFChars(cache_dir, options.accelerator_name); in Java_org_tensorflow_lite_nnapi_NnApiDelegate_createDelegate()
66 env->ReleaseStringUTFChars(model_token, options.accelerator_name); in Java_org_tensorflow_lite_nnapi_NnApiDelegate_createDelegate()
/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
Dnnapi_plugin.h56 if (nnapi_settings->accelerator_name() && in NnapiPlugin()
57 nnapi_settings->accelerator_name()->Length() != 0) { in NnapiPlugin()
58 accelerator_ = nnapi_settings->accelerator_name()->str(); in NnapiPlugin()
59 options_.accelerator_name = accelerator_.c_str(); in NnapiPlugin()
Dconfiguration_generated.h573 std::string accelerator_name; member
607 const flatbuffers::String *accelerator_name() const { in accelerator_name() function
640 verifier.VerifyString(accelerator_name()) && in Verify()
663 void add_accelerator_name(flatbuffers::Offset<flatbuffers::String> accelerator_name) { in add_accelerator_name()
664 fbb_.AddOffset(NNAPISettings::VT_ACCELERATOR_NAME, accelerator_name); in add_accelerator_name()
707 flatbuffers::Offset<flatbuffers::String> accelerator_name = 0,
724 builder_.add_accelerator_name(accelerator_name);
733 const char *accelerator_name = nullptr,
743 auto accelerator_name__ = accelerator_name ? _fbb.CreateString(accelerator_name) : 0;
2167 { auto _e = accelerator_name(); if (_e) _o->accelerator_name = _e->str(); } in UnPackTo()
[all …]
Dproto_to_flatbuffer.cc155 /*accelerator_name=*/builder->CreateString(settings.accelerator_name()), in ConvertNNAPISettings()
Dconfiguration.proto98 optional string accelerator_name = 1; field
/external/tensorflow/tensorflow/lite/tools/delegates/
Dnnapi_delegate_provider.cc103 std::string accelerator_name = in CreateTfLiteDelegate() local
105 if (!accelerator_name.empty()) { in CreateTfLiteDelegate()
106 options.accelerator_name = accelerator_name.c_str(); in CreateTfLiteDelegate()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_nnapi_failure_handling_test.cc59 const std::string& accelerator_name, in AddSubOpsAcceleratedModel() argument
63 options.accelerator_name = accelerator_name.c_str(); in AddSubOpsAcceleratedModel()
Dnnapi_delegate_device_selection_test.cc151 options.accelerator_name = "dsp"; in TEST_F()
220 AcceleratedModel(const NnApi* nnapi, const std::string& accelerator_name, in AcceleratedModel() argument
223 options.accelerator_name = accelerator_name.c_str(); in AcceleratedModel()
378 const std::string& accelerator_name, in AddSubOpsAcceleratedModel() argument
380 : MultiOpModel(), AcceleratedModel(nnapi, accelerator_name) { in AddSubOpsAcceleratedModel()
566 const std::string& accelerator_name, in HardSwishAddOpsAcceleratedModel() argument
568 : MultiOpModel(), AcceleratedModel(nnapi, accelerator_name) { in HardSwishAddOpsAcceleratedModel()
699 const NnApi* nnapi, std::string accelerator_name, const TensorData& input, in QuantizedWeightsConvolutionOpModel() argument
705 : SingleOpModel(), AcceleratedModel(nnapi, accelerator_name) {
813 const NnApi* nnapi, const std::string& accelerator_name, in LongIdentityModel() argument
[all …]
Dnnapi_delegate.h63 const char* accelerator_name = nullptr; member
218 std::string accelerator_name; member
Dnnapi_delegate.cc586 const char* device_name_ptr = delegate_options.accelerator_name; in ShouldUseTargetDevices()
617 const char* device_name_ptr = delegate_options.accelerator_name; in GetTargetDevices()
4821 if (options.accelerator_name) { in StatefulNnApiDelegate()
4822 delegate_data_.accelerator_name = options.accelerator_name; in StatefulNnApiDelegate()
4866 options.accelerator_name = delegate_data->accelerator_name.empty() in GetOptions()
4868 : delegate_data->accelerator_name.c_str(); in GetOptions()
5074 if (delegate_options.accelerator_name) { in DoPrepare()
Dnnapi_delegate_test.cc337 options.accelerator_name = "nnapi-reference"; in TEST()
358 options.accelerator_name = "foo"; in TEST()
395 options.accelerator_name = "nnapi-reference"; in TEST()