| /external/tensorflow/tensorflow/lite/nnapi/ |
| D | nnapi_implementation_test.cc | 21 const NnApi* nnapi = NnApiImplementation(); in TEST() local
22 EXPECT_NE(nnapi, nullptr); in TEST()
24 EXPECT_GT(nnapi->android_sdk_version, 0); in TEST()
25 if (nnapi.android_sdk_version < 27) { in TEST()
26 EXPECT_FALSE(nnapi->nnapi_exists); in TEST()
27 EXPECT_EQ(nnapi->ANeuralNetworksMemory_createFromFd, nullptr); in TEST()
28 EXPECT_EQ(nnapi->ANeuralNetworksMemory_free, nullptr); in TEST()
29 EXPECT_EQ(nnapi->ANeuralNetworksModel_create, nullptr); in TEST()
30 EXPECT_EQ(nnapi->ANeuralNetworksModel_free, nullptr); in TEST()
31 EXPECT_EQ(nnapi->ANeuralNetworksModel_finish, nullptr); in TEST()
[all …]
|
| D | nnapi_handler_test.cc | 25 namespace nnapi { namespace
39 const NnApi* nnapi = NnApiImplementation(); in TEST_F() local
48 ASSERT_THAT(nnapi->ANeuralNetworks_getDeviceCount, NotNull()); in TEST_F()
51 nnapi->ANeuralNetworks_getDeviceCount(&device_count); in TEST_F()
75 const NnApi* nnapi = NnApiImplementation(); in TEST_F() local
76 device_count_ptr = nnapi->ANeuralNetworks_getDeviceCount; in TEST_F()
85 EXPECT_THAT(nnapi->ANeuralNetworks_getDeviceCount(&device_count), Eq(1)); in TEST_F()
100 const NnApi* nnapi = NnApiImplementation(); in TEST_F() local
104 EXPECT_THAT(nnapi->ANeuralNetworks_getDeviceCount, IsNull()); in TEST_F()
105 EXPECT_THAT(nnapi->ANeuralNetworks_getDevice, IsNull()); in TEST_F()
[all …]
|
| D | nnapi_implementation.cc | 95 uint32_t CalculateAndroidSdkVersion(NnApi const& nnapi) { in CalculateAndroidSdkVersion() argument
97 bool has_10 = nnapi.ANeuralNetworksMemory_createFromFd != nullptr; in CalculateAndroidSdkVersion()
99 nnapi.ANeuralNetworksModel_relaxComputationFloat32toFloat16 != nullptr; in CalculateAndroidSdkVersion()
100 bool has_12 = nnapi.ANeuralNetworks_getDeviceCount != nullptr; in CalculateAndroidSdkVersion()
101 bool has_13 = nnapi.ANeuralNetworksCompilation_setTimeout != nullptr; in CalculateAndroidSdkVersion()
121 nnapi.name = reinterpret_cast<name##_fn>( \
125 nnapi.name = reinterpret_cast<name##_fn>( \
129 nnapi.name = reinterpret_cast<name##_fn>( \
133 NnApi nnapi = {}; in LoadNnApi() local
134 nnapi.android_sdk_version = 0; in LoadNnApi()
[all …]
|
| D | nnapi_implementation_disabled.cc | 18 static const NnApi nnapi = {}; in NnApiImplementation() local
19 return &nnapi; in NnApiImplementation()
|
| D | nnapi_handler.h | 23 namespace nnapi {
304 explicit NnApiHandler(NnApi* nnapi) : nnapi_(nnapi) { DCHECK(nnapi); } in NnApiHandler() argument
|
| D | nnapi_util.h | 25 namespace nnapi {
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | BUILD | 39 "//tensorflow/lite/nnapi:nnapi_implementation",
40 "//tensorflow/lite/nnapi:nnapi_lib",
41 "//tensorflow/lite/nnapi:nnapi_util",
74 "//tensorflow/lite/nnapi:nnapi_implementation",
75 "//tensorflow/lite/nnapi:nnapi_lib",
76 "//tensorflow/lite/nnapi:nnapi_util",
117 "//tensorflow/lite/nnapi:nnapi_handler",
118 "//tensorflow/lite/nnapi:nnapi_implementation",
119 "//tensorflow/lite/nnapi:nnapi_lib",
142 "//tensorflow/lite/nnapi:nnapi_implementation",
[all …]
|
| D | nnapi_delegate_kernel.h | 29 namespace nnapi {
124 explicit NNFreeModel(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeModel() argument
136 explicit NNFreeCompilation(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeCompilation() argument
148 explicit NNFreeExecution(const NnApi* nnapi) : nnapi_(nnapi) {} in NNFreeExecution() argument
161 NNMemory(const NnApi* nnapi, const char* name, size_t size);
248 explicit NNAPIDelegateKernel(const NnApi* nnapi) in NNAPIDelegateKernel() argument
250 nnapi_(nnapi), in NNAPIDelegateKernel()
|
| D | nnapi_delegate.h | 32 namespace nnapi {
37 using tflite::delegate::nnapi::NNAPIDelegateKernel;
136 explicit StatefulNnApiDelegate(const NnApi* nnapi);
151 StatefulNnApiDelegate(const NnApi* nnapi, Options options);
214 const NnApi* nnapi; member
255 explicit Data(const NnApi* nnapi);
304 TfLiteContext* context, TfLiteDelegate* delegate, const NnApi* nnapi,
|
| D | nnapi_delegate_mock_test.h | 37 namespace nnapi {
39 class NnApiMock : public ::tflite::nnapi::NnApiHandler {
41 explicit NnApiMock(NnApi* nnapi, int android_sdk_version = 29)
42 : ::tflite::nnapi::NnApiHandler(nnapi) { in NnApiHandler() argument
|
| D | nnapi_delegate_disabled.cc | 55 using ::tflite::delegate::nnapi::NNAPIDelegateKernel;
57 StatefulNnApiDelegate::Data::Data(const NnApi* nnapi) : nnapi(nnapi) {} in Data() argument
|
| D | nnapi_delegate_device_selection_test.cc | 44 void Init(const NnApi* nnapi, tflite::StatefulNnApiDelegate::Options options, in Init() argument
48 stateful_delegate_.reset(new StatefulNnApiDelegate(nnapi, options)); in Init()
79 : ::tflite::delegate::nnapi::NnApiDelegateMockTest {
81 ::tflite::delegate::nnapi::NnApiDelegateMockTest::SetUp(); in SetUp()
212 : ::tflite::delegate::nnapi::NnApiDelegateMockTest {};
220 AcceleratedModel(const NnApi* nnapi, const std::string& accelerator_name, in AcceleratedModel() argument
225 stateful_delegate_.reset(new StatefulNnApiDelegate(nnapi, options)); in AcceleratedModel()
230 AcceleratedModel(const NnApi* nnapi, bool disallow_nnapi_cpu, in AcceleratedModel() argument
235 stateful_delegate_.reset(new StatefulNnApiDelegate(nnapi, options)); in AcceleratedModel()
245 int axis_value, TensorType output_type, const NnApi* nnapi, in ArgMaxOpModel() argument
[all …]
|
| D | nnapi_delegate_errno_test.cc | 33 explicit SingleOpModelWithNNAPI(const NnApi* nnapi) { in SingleOpModelWithNNAPI() argument
35 stateful_delegate_.reset(new StatefulNnApiDelegate(nnapi, options_)); in SingleOpModelWithNNAPI()
52 FloatAddOpModel(const NnApi* nnapi, const TensorData& input1, in FloatAddOpModel() argument
56 : SingleOpModelWithNNAPI(nnapi) { in FloatAddOpModel()
89 struct NnApiErrnoTest : ::tflite::delegate::nnapi::NnApiDelegateMockTest {};
|
| D | nnapi_delegate_signed_quantization_test.cc | 43 void Init(const NnApi* nnapi) { in Init() argument
45 stateful_delegate_.reset(new StatefulNnApiDelegate(nnapi, options_)); in Init()
64 HybridFullyConnectedOpModel(const NnApi* nnapi, int units, int batches, in HybridFullyConnectedOpModel() argument
70 SingleOpModelWithNNAPI::Init(nnapi);
129 : ::tflite::delegate::nnapi::NnApiDelegateMockTest {
132 ::tflite::delegate::nnapi::NnApiDelegateMockTest::SetUp(); in SetUp()
229 const NnApi* nnapi, TfLiteRegistration* registration, in BaseConvolutionOpModel() argument
236 SingleOpModelWithNNAPI::Init(nnapi);
691 QuantizeOpModel(const NnApi* nnapi, const TensorData& input, in QuantizeOpModel() argument
693 SingleOpModelWithNNAPI::Init(nnapi); in QuantizeOpModel()
[all …]
|
| D | nnapi_delegate_nnapi_failure_handling_test.cc | 42 : ::tflite::delegate::nnapi::NnApiDelegateMockTest {};
58 const NnApi* nnapi, in AddSubOpsAcceleratedModel() argument
64 stateful_delegate_.reset(new StatefulNnApiDelegate(nnapi, options)); in AddSubOpsAcceleratedModel()
|
| D | nnapi_delegate.cc | 479 TfLiteStatus GetDeviceHandle(const NnApi* nnapi, TfLiteContext* context, in GetDeviceHandle() argument
486 nnapi->ANeuralNetworks_getDeviceCount(&num_devices); in GetDeviceHandle()
492 context, nnapi->ANeuralNetworks_getDevice(i, &device), in GetDeviceHandle()
496 context, nnapi->ANeuralNetworksDevice_getName(device, &buffer), in GetDeviceHandle()
509 nnapi::GetStringDeviceNamesList().c_str()); in GetDeviceHandle()
544 TfLiteContext* context, const NnApi* nnapi, in GetTargetSdkVersion() argument
547 *target_sdk_version = nnapi->android_sdk_version; in GetTargetSdkVersion()
553 nnapi->ANeuralNetworksDevice_getFeatureLevel(device_handle, in GetTargetSdkVersion()
565 (devices_sdk_version < nnapi->android_sdk_version)) { in GetTargetSdkVersion()
569 nnapi->android_sdk_version, devices_sdk_version); in GetTargetSdkVersion()
[all …]
|
| D | quant_lstm_sup_test.cc | 59 using tflite::delegate::nnapi::ExtractQuantLstmWeightsSubmatrix;
126 using tflite::delegate::nnapi::DecomposeQuantLstmWeightsTensor;
244 using tflite::delegate::nnapi::DecomposeBiasTensor;
|
| D | acceleration_test_util.h | 37 : min_android_sdk_version_{delegate::nnapi::kMinSdkVersionForNNAPI} {}; in NnapiAccelerationTestParams()
|
| /external/tensorflow/tensorflow/lite/ |
| D | Android.bp | 66 "delegates/nnapi/nnapi_delegate.cc",
67 "delegates/nnapi/quant_lstm_sup.cc",
81 "nnapi/nnapi_implementation.cc",
82 "nnapi/nnapi_util.cc",
|
| D | tflite_static.bp | 33 "delegates/nnapi/nnapi_delegate.cc",
34 "delegates/nnapi/quant_lstm_sup.cc",
157 "nnapi/nnapi_implementation.cc",
158 "nnapi/nnapi_util.cc",
|
| /external/tensorflow/tensorflow/lite/tools/delegates/ |
| D | BUILD | 86 "//tensorflow/lite/delegates/nnapi:nnapi_delegate",
87 "//tensorflow/lite/nnapi:nnapi_implementation",
88 "//tensorflow/lite/nnapi:nnapi_util",
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/ |
| D | BUILD | 126 "//tensorflow/lite/delegates/nnapi:nnapi_delegate",
147 "//tensorflow/lite/delegates/nnapi:nnapi_delegate",
148 "//tensorflow/lite/delegates/nnapi:nnapi_delegate_mock_test",
|
| /external/tensorflow/tensorflow/lite/tools/make/ |
| D | Makefile | 230 CORE_CC_ALL_SRCS += tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
231 CORE_CC_ALL_SRCS += tensorflow/lite/delegates/nnapi/quant_lstm_sup.cc
232 CORE_CC_ALL_SRCS += tensorflow/lite/nnapi/nnapi_implementation.cc
233 CORE_CC_ALL_SRCS += tensorflow/lite/nnapi/nnapi_util.cc
236 CORE_CC_ALL_SRCS += tensorflow/lite/delegates/nnapi/nnapi_delegate_disabled.cc
237 CORE_CC_ALL_SRCS += tensorflow/lite/nnapi/nnapi_implementation_disabled.cc
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 361 const NnApi* nnapi = NnApiImplementation(); in ExpectOpAcceleratedWithNnapi() local
362 if (nnapi && nnapi->nnapi_exists && in ExpectOpAcceleratedWithNnapi()
363 nnapi->android_sdk_version >= in ExpectOpAcceleratedWithNnapi()
|
| /external/tensorflow/tensorflow/lite/java/ |
| D | BUILD | 31 …]) + ["//tensorflow/lite/delegates/nnapi/java/src/main/java/org/tensorflow/lite/nnapi:nnapi_delega…
296 "src/test/java/org/tensorflow/lite/nnapi/NnApiDelegateTest.java",
303 test_class = "org.tensorflow.lite.nnapi.NnApiDelegateTest",
399 "src/test/java/org/tensorflow/lite/nnapi/NnApiDelegateTest.java",
472 "//tensorflow/lite/delegates/nnapi/java/src/main/native",
|