Lines Matching refs:RunTimeOperandInfo
55 OperationExecutionContext(const Operation* operation, RunTimeOperandInfo* operands) in OperationExecutionContext()
83 const RunTimeOperandInfo* getInputInfo(uint32_t index) const;
84 const RunTimeOperandInfo* getOutputInfo(uint32_t index) const;
85 RunTimeOperandInfo* getOutputInfo(uint32_t index);
88 RunTimeOperandInfo* operands;
93 const RunTimeOperandInfo* OperationExecutionContext::getInputInfo(uint32_t index) const { in getInputInfo()
98 const RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) const { in getOutputInfo()
103 RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) { in getOutputInfo()
154 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded()
517 static bool convertToNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertToNhwc()
561 static bool convertFromNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertFromNhwc()
610 RunTimeOperandInfo* operands) { in consumeOperationInputs()
628 static void freeUnusedSubgraphOperands(std::vector<RunTimeOperandInfo>* operands) { in freeUnusedSubgraphOperands()
654 std::vector<RunTimeOperandInfo> operands = initializeRunTimeInfo(model.main); in run()
682 int CpuExecutor::executeSubgraph(const Subgraph& subgraph, RunTimeOperandInfo* operands) { in executeSubgraph()
691 std::vector<RunTimeOperandInfo> CpuExecutor::initializeRunTimeInfo(const Subgraph& subgraph) { in initializeRunTimeInfo()
694 std::vector<RunTimeOperandInfo> operands(count); in initializeRunTimeInfo()
697 RunTimeOperandInfo& to = operands[i]; in initializeRunTimeInfo()
743 RunTimeOperandInfo* operands) { in updateForArguments()
748 RunTimeOperandInfo& to = operands[operandIndex]; in updateForArguments()
776 int CpuExecutor::executeOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeOperation()
854 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
855 const RunTimeOperandInfo& targetShape = operands[ins[1]]; in executeOperation()
857 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
871 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
875 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
878 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
935 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
939 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
942 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
999 const RunTimeOperandInfo& values = operands[ins[EmbeddingLookup::kValueTensor]]; in executeOperation()
1000 const RunTimeOperandInfo& lookups = operands[ins[EmbeddingLookup::kLookupTensor]]; in executeOperation()
1001 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()
1013 const RunTimeOperandInfo& lookups = operands[ins[HashtableLookup::kLookupTensor]]; in executeOperation()
1014 const RunTimeOperandInfo& keys = operands[ins[HashtableLookup::kKeyTensor]]; in executeOperation()
1015 const RunTimeOperandInfo& values = operands[ins[HashtableLookup::kValueTensor]]; in executeOperation()
1017 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation()
1018 RunTimeOperandInfo& hits = operands[outs[HashtableLookup::kHitsTensor]]; in executeOperation()
1029 RunTimeOperandInfo& output = operands[outs[LSHProjection::kOutputTensor]]; in executeOperation()
1037 const RunTimeOperandInfo& hash = operands[ins[LSHProjection::kHashTensor]]; in executeOperation()
1057 RunTimeOperandInfo& fwOutput = in executeOperation()
1068 RunTimeOperandInfo& bwOutput = in executeOperation()
1074 RunTimeOperandInfo& fwOutputActivationState = in executeOperation()
1077 RunTimeOperandInfo& fwOutputCellState = in executeOperation()
1079 RunTimeOperandInfo& bwOutputActivationState = in executeOperation()
1082 RunTimeOperandInfo& bwOutputCellState = in executeOperation()
1097 RunTimeOperandInfo& scratch = operands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1098 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1099 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1100 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1116 RunTimeOperandInfo& output = operands[outs[Multinomial::kOutputTensor]]; in executeOperation()
1130 RunTimeOperandInfo& hiddenStateOut = operands[outs[RNN::kHiddenStateOutTensor]]; in executeOperation()
1131 RunTimeOperandInfo& output = operands[outs[RNN::kOutputTensor]]; in executeOperation()
1141 RunTimeOperandInfo& stateOut = operands[outs[SVDF::kStateOutTensor]]; in executeOperation()
1142 RunTimeOperandInfo& output = operands[outs[SVDF::kOutputTensor]]; in executeOperation()
1156 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1157 const RunTimeOperandInfo& blockSize = operands[ins[1]]; in executeOperation()
1160 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1163 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1227 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1228 const RunTimeOperandInfo& blockSize = operands[ins[1]]; in executeOperation()
1229 const RunTimeOperandInfo& paddings = operands[ins[2]]; in executeOperation()
1232 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1235 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1305 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1306 const RunTimeOperandInfo& paddings = operands[ins[1]]; in executeOperation()
1308 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1345 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1347 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1358 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1359 const RunTimeOperandInfo& axis = operands[ins[1]]; in executeOperation()
1362 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1397 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1400 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1413 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1416 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1429 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1503 const RunTimeOperandInfo& in1 = operands[ins[0]]; in executeOperation()
1504 const RunTimeOperandInfo& in2 = operands[ins[1]]; in executeOperation()
1506 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1520 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1521 const RunTimeOperandInfo& filter = operands[ins[1]]; in executeOperation()
1522 const RunTimeOperandInfo& bias = operands[ins[2]]; in executeOperation()
1551 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1554 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1654 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1655 const RunTimeOperandInfo& multiples = operands[ins[1]]; in executeOperation()
1657 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1673 RunTimeOperandInfo& cellStateOut = in executeOperation()
1675 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation()
1690 const RunTimeOperandInfo& base = operands[ins[0]]; in executeOperation()
1691 const RunTimeOperandInfo& exponent = operands[ins[1]]; in executeOperation()
1693 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1735 static void setInfoExceptLifetime(RunTimeOperandInfo* to, const RunTimeOperandInfo& from) { in setInfoExceptLifetime()
1743 int CpuExecutor::executeIfOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeIfOperation()
1745 const RunTimeOperandInfo& condOperand = operands[operation.inputs[op::kCondBoolOperand]]; in executeIfOperation()
1754 const RunTimeOperandInfo& branchOperand = operands[operation.inputs[branchInputIndex]]; in executeIfOperation()
1756 std::vector<RunTimeOperandInfo> branchOperands = initializeRunTimeInfo(branchSubgraph); in executeIfOperation()
1781 int CpuExecutor::executeWhileOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeWhileOperation()
1783 const RunTimeOperandInfo& condModelOperand = operands[operation.inputs[op::kCondModelOperand]]; in executeWhileOperation()
1784 const RunTimeOperandInfo& bodyModelOperand = operands[operation.inputs[op::kBodyModelOperand]]; in executeWhileOperation()
1787 std::vector<RunTimeOperandInfo> condOperands = initializeRunTimeInfo(condSubgraph); in executeWhileOperation()
1788 std::vector<RunTimeOperandInfo> bodyOperands = initializeRunTimeInfo(bodySubgraph); in executeWhileOperation()
1836 RunTimeOperandInfo& condOutput = condOperands[condSubgraph.outputIndexes[0]]; in executeWhileOperation()
1874 RunTimeOperandInfo& info = bodyOperands[bodySubgraph.outputIndexes[i]]; in executeWhileOperation()
1896 RunTimeOperandInfo& outerOperand = operands[operation.outputs[i]]; in executeWhileOperation()
1897 RunTimeOperandInfo& innerOperand = condOperands[condSubgraph.inputIndexes[i]]; in executeWhileOperation()
1910 const std::vector<RunTimeOperandInfo>& operands) { in setOutputShapes()
1914 const RunTimeOperandInfo& from = operands[operandIndex]; in setOutputShapes()