Searched refs:operandIdx (Results 1 – 2 of 2) sorted by relevance
54 int SubGraphContext::addTensorFlatbuffer(TensorFlatbuffer tensor, int32_t operandIdx) { in addTensorFlatbuffer() argument58 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() argument70 CHECK(mOperandToTensorIdx[operandIdx] != -1); in addSubGraphInput()71 mInputTensors.push_back(mOperandToTensorIdx[operandIdx]); in addSubGraphInput()74 void SubGraphContext::addSubGraphOutput(int32_t operandIdx) { in addSubGraphOutput() argument75 CHECK(mOperandToTensorIdx[operandIdx] != -1); in addSubGraphOutput()76 mOutputTensors.push_back(mOperandToTensorIdx[operandIdx]); in addSubGraphOutput()[all …]
45 int addTensorFlatbuffer(TensorFlatbuffer tensor, int32_t operandIdx = -1);47 void addSubGraphInput(int32_t operandIdx);48 void addSubGraphOutput(int32_t operandIdx);52 int getTensorIdxFromOperandIdx(int operandIdx) const;69 Result<void> createTensorFlatbufferFromOperand(uint32_t operandIdx, bool makeSymmetric = false);