Searched defs:partitionTheWork (Results 1 - 2 of 2) sorted by relevance

/frameworks/ml/nn/runtime/
H A DExecutionPlan.cpp659 int ModelBuilder::partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, function in class:android::nn::ModelBuilder
670 VLOG(COMPILATION) << "ModelBuilder::partitionTheWork: deviceCount = " << deviceCount
692 VLOG(COMPILATION) << "ModelBuilder::partitionTheWork: only one best device: "
755 VLOG(COMPILATION) << "ModelBuilder::partitionTheWork: original model: ";
/frameworks/ml/nn/runtime/test/
H A DTestPartitioning.cpp270 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, function in class:PartitioningModel
272 return reinterpret_cast<ModelBuilder*>(getHandle())->partitionTheWork(
659 ASSERT_EQ(model.partitionTheWork(devicesA, ExecutePreference::PREFER_LOW_POWER, &planA),
676 ASSERT_EQ(model.partitionTheWork(devicesB, ExecutePreference::PREFER_LOW_POWER, &planB),
761 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan),

Completed in 40 milliseconds