Home
last modified time | relevance | path

Searched refs:mCompilation (Results 1 – 16 of 16) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/
DTestValidation.cpp201 ASSERT_EQ(ANeuralNetworksCompilation_create(mModel, &mCompilation), in SetUp()
206 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
210 ANeuralNetworksCompilation* mCompilation = nullptr; member in __anon437755720111::ValidationTestCompilation
218 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in SetUp()
220 ASSERT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in SetUp()
235 ASSERT_EQ(ANeuralNetworksBurst_create(mCompilation, &mBurst), ANEURALNETWORKS_NO_ERROR); in SetUp()
285 ASSERT_EQ(ANeuralNetworksCompilation_create(mModel, &mCompilation), in SetUp()
287 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in SetUp()
288 ASSERT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in SetUp()
313 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
[all …]
DTestNeuralNetworksWrapper.h268 int result = ANeuralNetworksCompilation_create(model->getHandle(), &mCompilation); in Compilation()
276 ~Compilation() { ANeuralNetworksCompilation_free(mCompilation); } in ~Compilation()
290 ANeuralNetworksCompilation_free(mCompilation);
291 mCompilation = other.mCompilation;
292 other.mCompilation = nullptr;
299 mCompilation, static_cast<int32_t>(preference))); in setPreference()
304 mCompilation, static_cast<int32_t>(priority))); in setPriority()
312 mCompilation, cacheDir.c_str(), token.data())); in setCaching()
315 Result finish() { return static_cast<Result>(ANeuralNetworksCompilation_finish(mCompilation)); } in finish()
321 mCompilation, index, alignment))); in getPreferredMemoryAlignmentForInput()
[all …]
DTestExecution.cpp615 mCompilation = reinterpret_cast<ANeuralNetworksCompilation*>(c); in TestCompilation()
645 model->getHandle(), mDevices.data(), mDevices.size(), &mCompilation); in TestIntrospectionCompilation()
664 mCompilation = TestIntrospectionCompilation(&mModel, kName); in ExecutionTestTemplate()
666 mCompilation = TestCompilation<DriverClass>(&mModel, kName, kForceErrorStatus); in ExecutionTestTemplate()
706 WrapperCompilation mCompilation; member in android::__anon3a757afb0111::ExecutionTestTemplate
758 ASSERT_EQ(mCompilation.finish(), WrapperResult::NO_ERROR); in TestWait()
762 WrapperExecution execution(&mCompilation); in TestWait()
790 WrapperExecution execution(&mCompilation); in TestWait()
822 WrapperExecution execution(&mCompilation); in TestWait()
852 WrapperExecution execution(&mCompilation); in TestWait()
DTestIntrospectionControl.cpp119 if (mCompilation) { in TearDown()
120 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
184 mDevices.size(), &mCompilation); in prepareForExecution()
188 EXPECT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in prepareForExecution()
189 EXPECT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in prepareForExecution()
203 ANeuralNetworksCompilation* mCompilation = nullptr; member in __anonb4e664590111::IntrospectionControlTest
245 CompilationBuilder* c = reinterpret_cast<CompilationBuilder*>(mCompilation); in TEST_F()
964 ASSERT_EQ(ANeuralNetworksBurst_create(mCompilation, &burst), ANEURALNETWORKS_NO_ERROR); in TEST_P()
1433 mDevices.size(), &mCompilation), in TEST_F()
1437 EXPECT_NE(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in TEST_F()
DTestGpuNnapi.cpp869 std::tie(result, mCompilation) = in initialize()
874 result = mCompilation.finish(); in initialize()
887 mExecution = std::make_unique<test_wrapper::Execution>(&mCompilation); in runInternal()
921 test_wrapper::Compilation mCompilation; member in android::nn::__anon23f5b2a90111::NnapiExecutor
DTestPartitioning.cpp948 mCompilation = reinterpret_cast<ANeuralNetworksCompilation*>(c); in PartitioningCompilation()
2287 std::optional<PartitioningCompilation> mCompilation; member in __anon135c357f0111::DynamicTemporariesTest
2343 ASSERT_TRUE(!mCompilation.has_value()); in compileModelAndComparePlan()
2358 mCompilation = PartitioningCompilation(&mModel.value(), devices); in compileModelAndComparePlan()
2359 ASSERT_EQ(mCompilation->setPartitioning(DeviceManager::kPartitioningWithoutFallback), in compileModelAndComparePlan()
2362 ASSERT_EQ(mCompilation->finish(), Result::NO_ERROR); in compileModelAndComparePlan()
2363 const ExecutionPlan& planA = mCompilation->getExecutionPlan(); in compileModelAndComparePlan()
2417 ASSERT_EQ(mCompilation->finish(), Result::OP_FAILED); in compileModelAndComparePlan()
2419 mCompilation = PartitioningCompilation(&mModel.value(), devices); in compileModelAndComparePlan()
2420 ASSERT_EQ(mCompilation->setPartitioning(DeviceManager::kPartitioningWithFallback), in compileModelAndComparePlan()
[all …]
DTestPartitioningRandom.cpp230 mCompilation = reinterpret_cast<ANeuralNetworksCompilation*>(c); in TestCompilation()
/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/
DFibonacciExtensionTest.cpp86 if (mCompilation) { in TearDown()
87 ANeuralNetworksCompilation_free(mCompilation); in TearDown()
113 mDevices.size(), &mCompilation), in prepareForExecution()
115 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_NO_ERROR); in prepareForExecution()
116 ASSERT_EQ(ANeuralNetworksExecution_create(mCompilation, &mExecution), in prepareForExecution()
125 ANeuralNetworksCompilation* mCompilation = nullptr; member in android::nn::__anon02ad936b0111::FibonacciExtensionTest
252 mDevices.size(), &mCompilation), in TEST_F()
254 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_BAD_DATA); in TEST_F()
263 mDevices.size(), &mCompilation), in TEST_F()
265 ASSERT_EQ(ANeuralNetworksCompilation_finish(mCompilation), ANEURALNETWORKS_BAD_DATA); in TEST_F()
[all …]
/packages/modules/NeuralNetworks/shim_and_sl/include/
DSupportLibraryWrapper.h367 ~Compilation() { mNnApi->getFL5()->ANeuralNetworksCompilation_free(mCompilation); } in ~Compilation()
382 mNnApi->getFL5()->ANeuralNetworksCompilation_free(mCompilation);
383 mCompilation = other.mCompilation;
384 other.mCompilation = nullptr;
391 mCompilation, static_cast<int32_t>(preference))); in setPreference()
396 mCompilation, static_cast<int32_t>(priority))); in setPriority()
401 mNnApi->getFL5()->ANeuralNetworksCompilation_setTimeout(mCompilation, durationNs)); in setTimeout()
409 mCompilation, cacheDir.c_str(), token.data())); in setCaching()
420 mCompilation, modelCacheFds.data(), modelCacheFds.size(), in setCachingFromFds()
439 mNnApi->getFL5()->ANeuralNetworksCompilation_finish(mCompilation)); in finish()
[all …]
DShimPreparedModel.h42 mCompilation(std::move(compilation)), in ShimPreparedModel()
71 const ::android::nn::sl_wrapper::Compilation& getCompilation() const { return mCompilation; } in getCompilation()
99 ::android::nn::sl_wrapper::Compilation mCompilation;
/packages/modules/NeuralNetworks/runtime/include/
DNeuralNetworksWrapper.h484 NNAPI_CALL(ANeuralNetworksCompilation_create(model->getHandle(), &mCompilation));
491 ~Compilation() { NNAPI_CALL(ANeuralNetworksCompilation_free(mCompilation)); }
505 NNAPI_CALL(ANeuralNetworksCompilation_free(mCompilation));
506 mCompilation = other.mCompilation;
507 other.mCompilation = nullptr;
514 mCompilation, static_cast<int32_t>(preference))));
519 mCompilation, static_cast<int32_t>(priority))));
527 mCompilation, cacheDir.c_str(), token.data())));
531 return static_cast<Result>(NNAPI_CALL(ANeuralNetworksCompilation_finish(mCompilation)));
538 mCompilation, index, alignment)));
[all …]
/packages/modules/NeuralNetworks/runtime/
DBurstBuilder.cpp34 : mCompilation(compilation), mBurstControllers(std::move(burstControllers)) {} in BurstBuilder()
46 return mCompilation; in getCompilation()
DExecutionBuilder.cpp154 : mCompilation(compilation), in ExecutionBuilder()
232 if (!memory->getValidator().validate(mCompilation, IOType::INPUT, index, type, offset, in setInputFromMemory()
313 if (!memory->getValidator().validate(mCompilation, IOType::OUTPUT, index, type, offset, in setOutputFromMemory()
341 if (!mCompilation->mExplicitDeviceList || (mCompilation->mDevices.size() != 1)) { in setMeasureTiming()
418 if (!mCompilation->mExplicitDeviceList || (mCompilation->mDevices.size() != 1)) { in setTimeoutDuration()
489 if (!mCompilation->mExplicitDeviceList || (mCompilation->mDevices.size() != 1)) { in addExtensionAttribute()
944 CHECK(!mCompilation->hasDynamicTemporaries()); in computeFencedInternal()
1017 if (!mCompilation->mExplicitDeviceList || (mCompilation->mDevices.size() != 1)) { in computeFenced()
DBurstBuilder.h53 const CompilationBuilder* mCompilation; variable
DExecutionBuilder.h116 const CompilationBuilder* getCompilation() const { return mCompilation; } in getCompilation()
181 const CompilationBuilder* mCompilation; variable
/packages/modules/NeuralNetworks/shim_and_sl/
DShimPreparedModel.cpp315 std::make_shared<::android::nn::sl_wrapper::Execution>(mNnapi.get(), &mCompilation); in executeFencedCommon()
412 std::make_shared<::android::nn::sl_wrapper::Execution>(mNnapi.get(), &mCompilation); in executeSynchronouslyCommon()
566 std::make_shared<::android::nn::sl_wrapper::Execution>(mNnapi.get(), &mCompilation); in createReusableExecution()