Home
last modified time | relevance | path

Searched refs:NnApi (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_kernel.h124 explicit NNFreeModel(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeModel()
131 const NnApi* nnapi_;
136 explicit NNFreeCompilation(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeCompilation()
143 const NnApi* nnapi_;
148 explicit NNFreeExecution(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeExecution()
155 const NnApi* nnapi_;
161 NNMemory(const NnApi* nnapi, const char* name, size_t size);
171 const NnApi* nnapi_;
248 explicit NNAPIDelegateKernel(const NnApi* nnapi) in NNAPIDelegateKernel()
317 const NnApi* nnapi_;
Dnnapi_delegate.h136 explicit StatefulNnApiDelegate(const NnApi* nnapi);
151 StatefulNnApiDelegate(const NnApi* nnapi, Options options);
214 const NnApi* nnapi;
255 explicit Data(const NnApi* nnapi);
304 TfLiteContext* context, TfLiteDelegate* delegate, const NnApi* nnapi,
Dnnapi_delegate_disabled.cc26 StatefulNnApiDelegate::StatefulNnApiDelegate(const NnApi* /* nnapi */) in StatefulNnApiDelegate() argument
32 StatefulNnApiDelegate::StatefulNnApiDelegate(const NnApi* /* nnapi */, in StatefulNnApiDelegate() argument
57 StatefulNnApiDelegate::Data::Data(const NnApi* nnapi) : nnapi(nnapi) {} in Data()
Dnnapi_delegate_mock_test.h41 explicit NnApiMock(NnApi* nnapi, int android_sdk_version = 29)
92 NnApi nnapi_;
Dnnapi_delegate_device_selection_test.cc44 void Init(const NnApi* nnapi, tflite::StatefulNnApiDelegate::Options options, in Init()
220 AcceleratedModel(const NnApi* nnapi, const std::string& accelerator_name, in AcceleratedModel()
230 AcceleratedModel(const NnApi* nnapi, bool disallow_nnapi_cpu, in AcceleratedModel()
245 int axis_value, TensorType output_type, const NnApi* nnapi, in ArgMaxOpModel()
252 int axis_value, TensorType output_type, const NnApi* nnapi, in ArgMaxOpModel()
377 const NnApi* nnapi, in AddSubOpsAcceleratedModel()
565 const NnApi* nnapi, in HardSwishAddOpsAcceleratedModel()
699 const NnApi* nnapi, std::string accelerator_name, const TensorData& input, in QuantizedWeightsConvolutionOpModel()
813 const NnApi* nnapi, const std::string& accelerator_name, in LongIdentityModel()
822 const NnApi* nnapi, int max_nnapi_partitions) in LongIdentityModel()
Dnnapi_delegate_errno_test.cc33 explicit SingleOpModelWithNNAPI(const NnApi* nnapi) { in SingleOpModelWithNNAPI()
52 FloatAddOpModel(const NnApi* nnapi, const TensorData& input1, in FloatAddOpModel()
Dnnapi_delegate_signed_quantization_test.cc43 void Init(const NnApi* nnapi) { in Init()
64 HybridFullyConnectedOpModel(const NnApi* nnapi, int units, int batches, in HybridFullyConnectedOpModel()
229 const NnApi* nnapi, TfLiteRegistration* registration, in BaseConvolutionOpModel()
691 QuantizeOpModel(const NnApi* nnapi, const TensorData& input, in QuantizeOpModel()
792 DequantizeOpModel(const NnApi* nnapi, TensorType type, in DequantizeOpModel()
Dnnapi_delegate_nnapi_failure_handling_test.cc58 const NnApi* nnapi, in AddSubOpsAcceleratedModel()
Dnnapi_delegate.cc479 TfLiteStatus GetDeviceHandle(const NnApi* nnapi, TfLiteContext* context, in GetDeviceHandle()
544 TfLiteContext* context, const NnApi* nnapi, in GetTargetSdkVersion()
584 const NnApi* nnapi, in ShouldUseTargetDevices()
610 const NnApi* nnapi, int* nnapi_errno, in GetTargetDevices()
654 NNMemory::NNMemory(const NnApi* nnapi, const char* name, size_t size) { in NNMemory()
666 NNMemory::NNMemory(const NnApi* /*nnapi*/, const char* /*name*/,
711 NNAPIOpBuilder(const NnApi* nnapi, TfLiteContext* context, in NNAPIOpBuilder()
1480 const NnApi* const nnapi_;
4782 StatefulNnApiDelegate::Data::Data(const NnApi* nnapi) : nnapi(nnapi) {} in Data()
4812 StatefulNnApiDelegate::StatefulNnApiDelegate(const NnApi* nnapi) in StatefulNnApiDelegate()
[all …]
/external/tensorflow/tensorflow/lite/nnapi/
Dnnapi_handler_test.cc31 void ExpectEquals(const NnApi& left, const NnApi& right);
39 const NnApi* nnapi = NnApiImplementation(); in TEST_F()
56 NnApi nnapi_orig_copy = *NnApiImplementation(); in TEST_F()
75 const NnApi* nnapi = NnApiImplementation(); in TEST_F()
100 const NnApi* nnapi = NnApiImplementation(); in TEST_F()
142 const NnApi* nnapi = NnApiImplementation(); in TEST_F()
175 void ExpectEquals(const NnApi& left, const NnApi& right) { in ExpectEquals()
Dnnapi_implementation_disabled.cc17 const NnApi* NnApiImplementation() { in NnApiImplementation()
18 static const NnApi nnapi = {}; in NnApiImplementation()
Dnnapi_implementation.cc95 uint32_t CalculateAndroidSdkVersion(NnApi const& nnapi) { in CalculateAndroidSdkVersion()
132 const NnApi LoadNnApi() { in LoadNnApi()
133 NnApi nnapi = {}; in LoadNnApi()
305 const NnApi* NnApiImplementation() { in NnApiImplementation()
306 static const NnApi nnapi = LoadNnApi(); in NnApiImplementation()
Dnnapi_handler.cc34 const NnApi* NnApiPassthroughInstance() { in NnApiPassthroughInstance()
35 static const NnApi orig_nnapi_copy = *NnApiImplementation(); in NnApiPassthroughInstance()
44 static NnApiHandler handler{const_cast<NnApi*>(NnApiImplementation())}; in Instance()
Dnnapi_handler.h301 const NnApi* GetNnApi() { return nnapi_; } in GetNnApi()
304 explicit NnApiHandler(NnApi* nnapi) : nnapi_(nnapi) { DCHECK(nnapi); } in NnApiHandler()
306 NnApi* nnapi_;
333 const NnApi* NnApiPassthroughInstance();
Dnnapi_implementation.h24 struct NnApi { struct
1667 const NnApi* NnApiImplementation();
Dnnapi_implementation_test.cc21 const NnApi* nnapi = NnApiImplementation(); in TEST()
/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
Dnnapi_plugin_test.cc63 nnapi_ = const_cast<NnApi*>(NnApiImplementation()); in SetUp()
120 NnApi* nnapi_;
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.cc361 const NnApi* nnapi = NnApiImplementation(); in ExpectOpAcceleratedWithNnapi()