/external/tensorflow/tensorflow/lite/delegates/nnapi/java/src/main/native/ |
D | nnapi_delegate_jni.cc | 26 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/ |
D | nnapi_plugin.h | 56 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()
|
D | configuration_generated.h | 573 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 …]
|
D | proto_to_flatbuffer.cc | 155 /*accelerator_name=*/builder->CreateString(settings.accelerator_name()), in ConvertNNAPISettings()
|
D | configuration.proto | 98 optional string accelerator_name = 1; field
|
/external/tensorflow/tensorflow/lite/tools/delegates/ |
D | nnapi_delegate_provider.cc | 103 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/ |
D | nnapi_delegate_nnapi_failure_handling_test.cc | 59 const std::string& accelerator_name, in AddSubOpsAcceleratedModel() argument 63 options.accelerator_name = accelerator_name.c_str(); in AddSubOpsAcceleratedModel()
|
D | nnapi_delegate_device_selection_test.cc | 151 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 …]
|
D | nnapi_delegate.h | 63 const char* accelerator_name = nullptr; member 218 std::string accelerator_name; member
|
D | nnapi_delegate.cc | 586 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()
|
D | nnapi_delegate_test.cc | 337 options.accelerator_name = "nnapi-reference"; in TEST() 358 options.accelerator_name = "foo"; in TEST() 395 options.accelerator_name = "nnapi-reference"; in TEST()
|