/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_kernel.h | 124 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_;
|
D | nnapi_delegate.h | 136 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,
|
D | nnapi_delegate_disabled.cc | 26 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()
|
D | nnapi_delegate_mock_test.h | 41 explicit NnApiMock(NnApi* nnapi, int android_sdk_version = 29) 92 NnApi nnapi_;
|
D | nnapi_delegate_device_selection_test.cc | 44 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()
|
D | nnapi_delegate_errno_test.cc | 33 explicit SingleOpModelWithNNAPI(const NnApi* nnapi) { in SingleOpModelWithNNAPI() 52 FloatAddOpModel(const NnApi* nnapi, const TensorData& input1, in FloatAddOpModel()
|
D | nnapi_delegate_signed_quantization_test.cc | 43 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()
|
D | nnapi_delegate_nnapi_failure_handling_test.cc | 58 const NnApi* nnapi, in AddSubOpsAcceleratedModel()
|
D | nnapi_delegate.cc | 479 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/ |
D | nnapi_handler_test.cc | 31 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()
|
D | nnapi_implementation_disabled.cc | 17 const NnApi* NnApiImplementation() { in NnApiImplementation() 18 static const NnApi nnapi = {}; in NnApiImplementation()
|
D | nnapi_implementation.cc | 95 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()
|
D | nnapi_handler.cc | 34 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()
|
D | nnapi_handler.h | 301 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();
|
D | nnapi_implementation.h | 24 struct NnApi { struct 1667 const NnApi* NnApiImplementation();
|
D | nnapi_implementation_test.cc | 21 const NnApi* nnapi = NnApiImplementation(); in TEST()
|
/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/ |
D | nnapi_plugin_test.cc | 63 nnapi_ = const_cast<NnApi*>(NnApiImplementation()); in SetUp() 120 NnApi* nnapi_;
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | test_util.cc | 361 const NnApi* nnapi = NnApiImplementation(); in ExpectOpAcceleratedWithNnapi()
|