Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestPartitioning.cpp577 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork() function in __anon83dd480d0111::PartitioningModel
579 return reinterpret_cast<ModelBuilder*>(getHandle())->partitionTheWork( in partitionTheWork()
1194 ASSERT_EQ(model.partitionTheWork(devicesA, ExecutePreference::PREFER_LOW_POWER, &planA), in TEST_F()
1205 ASSERT_EQ(model.partitionTheWork(devicesC, ExecutePreference::PREFER_LOW_POWER, &planC), in TEST_F()
1216 ASSERT_EQ(model.partitionTheWork(devicesB, ExecutePreference::PREFER_LOW_POWER, &planB), in TEST_F()
1282 ASSERT_EQ(model.partitionTheWork(devicesA, ExecutePreference::PREFER_LOW_POWER, &planA), in TEST_F()
1294 ASSERT_EQ(model.partitionTheWork(devicesB, ExecutePreference::PREFER_LOW_POWER, &planB), in TEST_F()
1382 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1420 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1562 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
[all …]
/frameworks/ml/nn/runtime/
DCompilationBuilder.cpp55 int n = mModel->partitionTheWork(mDevices, mPreference, &mPlan); in finish()
DModelBuilder.h102 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, uint32_t preference,
DExecutionPlan.cpp964 int ModelBuilder::partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork() function in android::nn::ModelBuilder