Lines Matching refs:ModelBuilder
105 int compile(const Device& device, const ModelBuilder& model, int executionPreference, in compile()
139 int copyOperandExtraParams(ModelBuilder& model, uint32_t toOperandIndex, in copyOperandExtraParams()
172 OperandTracker(const ModelBuilder* model, OperationReadyCallback cb);
179 const ModelBuilder* mModel;
184 OperandTracker::OperandTracker(const ModelBuilder* model, OperationReadyCallback cb) in OperandTracker()
470 const ModelBuilder& sourceModel = *getSourceModel(); in addOperand()
540 const ModelBuilder* model = sourceModel.getReferencedModel(operand); in addOperand()
732 const ModelBuilder* ExecutionStep::getSourceModel() const { in getSourceModel()
784 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel()
923 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish()
927 const ModelBuilder* sourceModel = sourceModels->getModel(sourceOperandIndex.first); in finish()
1015 const ModelBuilder* sourceModel = sourceModels->getModel(sourceOperandIndex.first); in findControlFlowBoundaryConstants()
1939 const ModelBuilder* model) { in becomeSingleStep()
2135 int ModelBuilder::partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork()
2152 int ModelBuilder::partitionTheWorkInternal(uint32_t sourceModelIndex, in partitionTheWorkInternal()
2278 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in partitionTheWorkInternal()
2279 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in partitionTheWorkInternal()
2335 const ModelBuilder* condModel = getReferencedModel(condOperand); in partitionTheWorkInternal()
2336 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in partitionTheWorkInternal()
2398 float ModelBuilder::getPerformance(uint32_t preference, in getPerformance()
2415 float ModelBuilder::getPerformance(uint32_t preference, const std::shared_ptr<Device> device, in getPerformance()
2427 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in getPerformance()
2428 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in getPerformance()
2438 const ModelBuilder* condModel = getReferencedModel(condOperand); in getPerformance()
2439 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in getPerformance()
2467 bool ModelBuilder::isControlFlowOperationWithOperandOfUnknownSize(uint32_t operationIndex) const { in isControlFlowOperationWithOperandOfUnknownSize()
2468 auto containsUnknownSize = [](const ModelBuilder* model, in isControlFlowOperationWithOperandOfUnknownSize()
2484 const ModelBuilder* thenModel = getReferencedModel(thenOperand); in isControlFlowOperationWithOperandOfUnknownSize()
2485 const ModelBuilder* elseModel = getReferencedModel(elseOperand); in isControlFlowOperationWithOperandOfUnknownSize()
2498 const ModelBuilder* condModel = getReferencedModel(condOperand); in isControlFlowOperationWithOperandOfUnknownSize()
2499 const ModelBuilder* bodyModel = getReferencedModel(bodyOperand); in isControlFlowOperationWithOperandOfUnknownSize()
2512 bool ModelBuilder::supportedByControlFlowInterpreter(uint32_t operationIndex) const { in supportedByControlFlowInterpreter()
2538 int ModelBuilder::findBestDeviceForEachOperation( in findBestDeviceForEachOperation()