| /frameworks/ml/nn/runtime/test/ |
| D | TestValidation.cpp | 190 ASSERT_EQ(ANeuralNetworksCompilation_create(mModel, &mCompilation), in SetUp()
195 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
199 ANeuralNetworksCompilation* mCompilation = nullptr; member in __anon1fd66c800111::ValidationTestCompilation
207 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in SetUp()
209 ASSERT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in SetUp()
224 ASSERT_EQ(ANeuralNetworksBurst_create(mCompilation, &mBurst), ANEURALNETWORKS_NO_ERROR); in SetUp()
268 ASSERT_EQ(ANeuralNetworksCompilation_create(mModel, &mCompilation), in SetUp()
270 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in SetUp()
271 ASSERT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in SetUp()
296 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
[all …]
|
| D | TestNeuralNetworksWrapper.h | 262 int result = ANeuralNetworksCompilation_create(model->getHandle(), &mCompilation); in Compilation()
270 ~Compilation() { ANeuralNetworksCompilation_free(mCompilation); } in ~Compilation()
284 ANeuralNetworksCompilation_free(mCompilation);
285 mCompilation = other.mCompilation;
286 other.mCompilation = nullptr;
293 mCompilation, static_cast<int32_t>(preference))); in setPreference()
298 mCompilation, static_cast<int32_t>(priority))); in setPriority()
306 mCompilation, cacheDir.c_str(), token.data())); in setCaching()
309 Result finish() { return static_cast<Result>(ANeuralNetworksCompilation_finish(mCompilation)); } in finish()
311 ANeuralNetworksCompilation* getHandle() const { return mCompilation; } in getHandle()
[all …]
|
| D | TestExecution.cpp | 580 mCompilation = reinterpret_cast<ANeuralNetworksCompilation*>(c); in TestCompilation()
610 model->getHandle(), mDevices.data(), mDevices.size(), &mCompilation); in TestIntrospectionCompilation()
629 mCompilation = TestIntrospectionCompilation(&mModel, kName); in ExecutionTestTemplate()
631 mCompilation = TestCompilation<DriverClass>(&mModel, kName, kForceErrorStatus); in ExecutionTestTemplate()
663 WrapperCompilation mCompilation; member in android::__anon50a6e8490111::ExecutionTestTemplate
702 ASSERT_EQ(mCompilation.finish(), Result::NO_ERROR); in TestWait()
713 WrapperExecution execution(&mCompilation); in TestWait()
736 WrapperExecution execution(&mCompilation); in TestWait()
769 WrapperExecution execution(&mCompilation); in TestWait()
792 WrapperExecution execution(&mCompilation); in TestWait()
|
| D | TestIntrospectionControl.cpp | 113 if (mCompilation) { in TearDown()
114 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
178 mDevices.size(), &mCompilation); in prepareForExecution()
182 EXPECT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in prepareForExecution()
183 EXPECT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in prepareForExecution()
197 ANeuralNetworksCompilation* mCompilation = nullptr; member in __anon736024e70111::IntrospectionControlTest
239 CompilationBuilder* c = reinterpret_cast<CompilationBuilder*>(mCompilation); in TEST_F()
939 ASSERT_EQ(ANeuralNetworksBurst_create(mCompilation, &burst), ANEURALNETWORKS_NO_ERROR); in TEST_P()
1231 mDevices.size(), &mCompilation), in TEST_F()
1235 EXPECT_NE(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in TEST_F()
|
| D | TestPartitioningRandom.cpp | 217 mCompilation = reinterpret_cast<ANeuralNetworksCompilation*>(c); in TestCompilation()
|
| D | TestPartitioning.cpp | 877 mCompilation = reinterpret_cast<ANeuralNetworksCompilation*>(c); in PartitioningCompilation()
|
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciExtensionTest.cpp | 88 if (mCompilation) { in TearDown()
89 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
110 mDevices.size(), &mCompilation), in prepareForExecution()
112 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in prepareForExecution()
113 ASSERT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in prepareForExecution()
119 ANeuralNetworksCompilation* mCompilation = nullptr; member in android::nn::__anon7b3c0b790111::FibonacciExtensionTest
246 mDevices.size(), &mCompilation), in TEST_F()
248 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_BAD_DATA); in TEST_F()
257 mDevices.size(), &mCompilation), in TEST_F()
259 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_BAD_DATA); in TEST_F()
[all …]
|
| /frameworks/ml/nn/runtime/include/ |
| D | NeuralNetworksWrapper.h | 351 int result = ANeuralNetworksCompilation_create(model->getHandle(), &mCompilation); in Compilation()
357 ~Compilation() { ANeuralNetworksCompilation_free(mCompilation); } in ~Compilation()
371 ANeuralNetworksCompilation_free(mCompilation);
372 mCompilation = other.mCompilation;
373 other.mCompilation = nullptr;
380 mCompilation, static_cast<int32_t>(preference))); in setPreference()
385 mCompilation, static_cast<int32_t>(priority))); in setPriority()
393 mCompilation, cacheDir.c_str(), token.data())); in setCaching()
396 Result finish() { return static_cast<Result>(ANeuralNetworksCompilation_finish(mCompilation)); } in finish()
398 ANeuralNetworksCompilation* getHandle() const { return mCompilation; } in getHandle()
[all …]
|
| /frameworks/ml/nn/runtime/ |
| D | BurstBuilder.cpp | 29 : mCompilation(compilation), mBurstControllers(std::move(burstControllers)) {} in BurstBuilder()
41 return mCompilation; in getCompilation()
|
| D | BurstBuilder.h | 53 const CompilationBuilder* mCompilation; variable
|
| D | ExecutionBuilder.cpp | 93 : mCompilation(compilation), in ExecutionBuilder()
182 if (!memory->getValidator().validate(mCompilation, IOType::INPUT, index, type, offset, in setInputFromMemory()
259 if (!memory->getValidator().validate(mCompilation, IOType::OUTPUT, index, type, offset, in setOutputFromMemory()
285 if (!mCompilation->mExplicitDeviceList || (mCompilation->mDevices.size() != 1)) { in setMeasureTiming()
367 if (!mCompilation->mExplicitDeviceList || (mCompilation->mDevices.size() != 1)) { in setTimeoutDuration()
738 if (!mCompilation->mExplicitDeviceList || (mCompilation->mDevices.size() != 1)) { in computeFenced()
|
| D | ExecutionBuilder.h | 98 const CompilationBuilder* getCompilation() const { return mCompilation; } in getCompilation()
134 const CompilationBuilder* mCompilation; variable
|