• Home
  • Raw
  • Download

Lines Matching refs:mainModel

744 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize,  in finishStepModel()  argument
757 mStepModel.relaxComputationFloat32toFloat16(mainModel->isComputationFloat32RelaxedToFloat16()); in finishStepModel()
780 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finishStepModel()
781 mainModelOperandToInputIndex[mainModel->getInputOperandIndex(i)] = i; in finishStepModel()
784 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finishStepModel()
785 mainModelOperandToOutputIndex[mainModel->getOutputOperandIndex(i)] = i; in finishStepModel()
880 const ModelBuilder* mainModel = sourceModels->getModel(kMainModelInSourceModels); in finish() local
897 int n = step->finishStepModel(mainModel, &stepHasDynamicTemporaries, in finish()
949 for (uint32_t i = 0, n = mainModel->inputCount(); i < n; ++i) { in finish()
950 SourceOperandIndex index(kMainModelInSourceModels, mainModel->getInputOperandIndex(i)); in finish()
953 for (uint32_t i = 0, n = mainModel->outputCount(); i < n; ++i) { in finish()
954 SourceOperandIndex index(kMainModelInSourceModels, mainModel->getOutputOperandIndex(i)); in finish()
1990 const auto* mainModel = getSourceModels().getModel(kMainModelInSourceModels); in getInputSourceOperand() local
1991 CHECK_LT(index, mainModel->inputCount()); in getInputSourceOperand()
1992 const auto operandIndex = mainModel->getInputOperandIndex(index); in getInputSourceOperand()
1997 const auto* mainModel = getSourceModels().getModel(kMainModelInSourceModels); in getOutputSourceOperand() local
1998 CHECK_LT(index, mainModel->outputCount()); in getOutputSourceOperand()
1999 const auto operandIndex = mainModel->getOutputOperandIndex(index); in getOutputSourceOperand()