Lines Matching refs:ModelBuilder
134 int compileModelAndCache(const std::shared_ptr<Device>& device, const ModelBuilder* model, in compileModelAndCache()
159 int compile(std::shared_ptr<Device> device, const ModelBuilder* model, int32_t executionPreference, in compile()
178 int copyOperandExtraParams(ModelBuilder& model, uint32_t toOperandIndex, in copyOperandExtraParams()
214 OperandTracker(const ModelBuilder* model, OperationReadyCallback cb);
221 const ModelBuilder* mModel;
226 OperandTracker::OperandTracker(const ModelBuilder* model, OperationReadyCallback cb) : in OperandTracker()
272 const ModelBuilder& fromModel, OperandKind kind) { in addOperand()
370 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) { in addOperation()
493 int ExecutionStep::finishSubModel(const ModelBuilder* fromModel, bool* hasOutputOfUnknownSize, in finishSubModel()
591 int ExecutionPlan::CompoundBody::finish(const ModelBuilder* fromModel, in finish()
611 int ExecutionPlan::SimpleBody::finish([[maybe_unused]] const ModelBuilder* fromModel, in finish()
621 int ExecutionPlan::finish(const ModelBuilder* fromModel, int32_t executionPreference) { in finish()
709 const ModelBuilder* fromModel = executionBuilder->getModel(); in makeController()
897 const ModelBuilder* model) { in becomeSingleStep()
964 int ModelBuilder::partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork()
1054 PerformanceInfo ModelBuilder::getPerformanceInfo(const std::shared_ptr<Device> device, in getPerformanceInfo()
1161 PlanModelSlicer(const ModelBuilder* model);
1242 PlanModelSlicer::PlanModelSlicer(const ModelBuilder* model) { in PlanModelSlicer()
1555 int ModelBuilder::findBestDeviceForEachOperation( in findBestDeviceForEachOperation()