Lines Matching refs:mainModel
479 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel() argument
492 mStepModel.relaxComputationFloat32toFloat16(mainModel->isComputationFloat32RelaxedToFloat16()); in finishStepModel()
506 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finishStepModel()
507 mainModelOperandToInputIndex[mainModel->getInputOperandIndex(i)] = i; in finishStepModel()
510 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finishStepModel()
511 mainModelOperandToOutputIndex[mainModel->getOutputOperandIndex(i)] = i; in finishStepModel()
616 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish() local
632 int n = step->finishStepModel(mainModel, &mHasStepModelOutputOfUnknownSize, in finish()
666 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finish()
667 SourceOperandIndex index(kMainModelInSourceModels, mainModel->getInputOperandIndex(i)); in finish()
670 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finish()
671 SourceOperandIndex index(kMainModelInSourceModels, mainModel->getOutputOperandIndex(i)); in finish()