Lines Matching refs:operandIdx
54 int SubGraphContext::addTensorFlatbuffer(TensorFlatbuffer tensor, int32_t operandIdx) { in addTensorFlatbuffer() argument
58 if (operandIdx >= 0) { in addTensorFlatbuffer()
59 CHECK(mOperandToTensorIdx[operandIdx] == -1); in addTensorFlatbuffer()
60 mOperandToTensorIdx[operandIdx] = tensorIdx; in addTensorFlatbuffer()
69 void SubGraphContext::addSubGraphInput(int32_t operandIdx) { in addSubGraphInput() argument
70 CHECK(mOperandToTensorIdx[operandIdx] != -1); in addSubGraphInput()
71 mInputTensors.push_back(mOperandToTensorIdx[operandIdx]); in addSubGraphInput()
74 void SubGraphContext::addSubGraphOutput(int32_t operandIdx) { in addSubGraphOutput() argument
75 CHECK(mOperandToTensorIdx[operandIdx] != -1); in addSubGraphOutput()
76 mOutputTensors.push_back(mOperandToTensorIdx[operandIdx]); in addSubGraphOutput()
104 int SubGraphContext::getTensorIdxFromOperandIdx(int operandIdx) const { in getTensorIdxFromOperandIdx()
105 return mOperandToTensorIdx[operandIdx]; in getTensorIdxFromOperandIdx()
147 Result<void> SubGraphContext::createTensorFlatbufferFromOperand(uint32_t operandIdx, in createTensorFlatbufferFromOperand() argument
152 if (mOperandToTensorIdx[operandIdx] != -1) return {}; in createTensorFlatbufferFromOperand()
154 const Operand& operand = mSubgraph->operands[operandIdx]; in createTensorFlatbufferFromOperand()
208 addTensorFlatbuffer(tensor, operandIdx); in createTensorFlatbufferFromOperand()