HomeSort by relevance Sort by last modified time
    Searched refs:partitionTheWork (Results 1 - 4 of 4) sorted by null

  /frameworks/ml/nn/runtime/
CompilationBuilder.cpp 50 int n = mModel->partitionTheWork(devices, mPreference, &mPlan);
ModelBuilder.h 84 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices,
88 // TODO: move partitionTheWork, findBestDeviceForEachOperation,
ExecutionPlan.cpp 766 int ModelBuilder::partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices,
777 VLOG(COMPILATION) << "ModelBuilder::partitionTheWork: deviceCount = " << deviceCount
    [all...]
  /frameworks/ml/nn/runtime/test/
TestPartitioning.cpp 334 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices,
336 return reinterpret_cast<ModelBuilder*>(getHandle())->partitionTheWork(
    [all...]

Completed in 63 milliseconds