Home
last modified time | relevance | path

Searched refs:partitionTheWork (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/runtime/
DCompilationBuilder.cpp50 int n = mModel->partitionTheWork(devices, mPreference, &mPlan); in finish()
DModelBuilder.h84 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices,
DExecutionPlan.cpp766 int ModelBuilder::partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork() function in android::nn::ModelBuilder
/frameworks/ml/nn/runtime/test/
DTestPartitioning.cpp334 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork() function in __anon83dd480d0111::PartitioningModel
336 return reinterpret_cast<ModelBuilder*>(getHandle())->partitionTheWork( in partitionTheWork()
772 ASSERT_EQ(model.partitionTheWork(devicesA, ExecutePreference::PREFER_LOW_POWER, &planA), in TEST_F()
787 ASSERT_EQ(model.partitionTheWork(devicesC, ExecutePreference::PREFER_LOW_POWER, &planC), in TEST_F()
804 ASSERT_EQ(model.partitionTheWork(devicesB, ExecutePreference::PREFER_LOW_POWER, &planB), in TEST_F()
893 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1045 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1164 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()