Lines Matching refs:RunTimeOperandInfo
69 OperationExecutionContext(const Operation* operation, RunTimeOperandInfo* operands) in OperationExecutionContext()
97 const RunTimeOperandInfo* getInputInfo(uint32_t index) const;
98 const RunTimeOperandInfo* getOutputInfo(uint32_t index) const;
99 RunTimeOperandInfo* getOutputInfo(uint32_t index);
102 RunTimeOperandInfo* operands;
107 const RunTimeOperandInfo* OperationExecutionContext::getInputInfo(uint32_t index) const { in getInputInfo()
112 const RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) const { in getOutputInfo()
117 RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) { in getOutputInfo()
168 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded()
434 static bool convertToNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertToNhwc()
478 static bool convertFromNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertFromNhwc()
528 RunTimeOperandInfo* operands) { in consumeOperationInputs()
546 static void freeUnusedSubgraphOperands(std::vector<RunTimeOperandInfo>* operands) { in freeUnusedSubgraphOperands()
572 std::vector<RunTimeOperandInfo> operands = initializeRunTimeInfo(model.main); in run()
600 int CpuExecutor::executeSubgraph(const Model::Subgraph& subgraph, RunTimeOperandInfo* operands) { in executeSubgraph()
609 std::vector<RunTimeOperandInfo> CpuExecutor::initializeRunTimeInfo( in initializeRunTimeInfo()
613 std::vector<RunTimeOperandInfo> operands(count); in initializeRunTimeInfo()
618 RunTimeOperandInfo& to = operands[i]; in initializeRunTimeInfo()
669 RunTimeOperandInfo* operands) { in updateForArguments()
674 RunTimeOperandInfo& to = operands[operandIndex]; in updateForArguments()
719 [[maybe_unused]] RunTimeOperandInfo* operands) { in executeOperation()
799 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
800 const RunTimeOperandInfo& targetShape = operands[ins[1]]; in executeOperation()
802 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
816 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
820 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
823 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
880 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
884 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
887 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
944 const RunTimeOperandInfo& values = operands[ins[EmbeddingLookup::kValueTensor]]; in executeOperation()
945 const RunTimeOperandInfo& lookups = operands[ins[EmbeddingLookup::kLookupTensor]]; in executeOperation()
946 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()
958 const RunTimeOperandInfo& lookups = operands[ins[HashtableLookup::kLookupTensor]]; in executeOperation()
959 const RunTimeOperandInfo& keys = operands[ins[HashtableLookup::kKeyTensor]]; in executeOperation()
960 const RunTimeOperandInfo& values = operands[ins[HashtableLookup::kValueTensor]]; in executeOperation()
962 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation()
963 RunTimeOperandInfo& hits = operands[outs[HashtableLookup::kHitsTensor]]; in executeOperation()
974 RunTimeOperandInfo& output = operands[outs[LSHProjection::kOutputTensor]]; in executeOperation()
982 const RunTimeOperandInfo& hash = operands[ins[LSHProjection::kHashTensor]]; in executeOperation()
1002 RunTimeOperandInfo& fwOutput = in executeOperation()
1013 RunTimeOperandInfo& bwOutput = in executeOperation()
1019 RunTimeOperandInfo& fwOutputActivationState = in executeOperation()
1022 RunTimeOperandInfo& fwOutputCellState = in executeOperation()
1024 RunTimeOperandInfo& bwOutputActivationState = in executeOperation()
1027 RunTimeOperandInfo& bwOutputCellState = in executeOperation()
1042 RunTimeOperandInfo& scratch = operands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1043 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1044 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1045 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1061 RunTimeOperandInfo& output = operands[outs[Multinomial::kOutputTensor]]; in executeOperation()
1075 RunTimeOperandInfo& hiddenStateOut = operands[outs[RNN::kHiddenStateOutTensor]]; in executeOperation()
1076 RunTimeOperandInfo& output = operands[outs[RNN::kOutputTensor]]; in executeOperation()
1086 RunTimeOperandInfo& stateOut = operands[outs[SVDF::kStateOutTensor]]; in executeOperation()
1087 RunTimeOperandInfo& output = operands[outs[SVDF::kOutputTensor]]; in executeOperation()
1101 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1102 const RunTimeOperandInfo& blockSize = operands[ins[1]]; in executeOperation()
1105 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1108 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1172 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1173 const RunTimeOperandInfo& blockSize = operands[ins[1]]; in executeOperation()
1174 const RunTimeOperandInfo& paddings = operands[ins[2]]; in executeOperation()
1177 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1180 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1250 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1251 const RunTimeOperandInfo& paddings = operands[ins[1]]; in executeOperation()
1253 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1290 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1292 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1303 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1304 const RunTimeOperandInfo& axis = operands[ins[1]]; in executeOperation()
1307 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1342 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1345 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1358 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1361 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1374 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1448 const RunTimeOperandInfo& in1 = operands[ins[0]]; in executeOperation()
1449 const RunTimeOperandInfo& in2 = operands[ins[1]]; in executeOperation()
1451 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1465 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1466 const RunTimeOperandInfo& filter = operands[ins[1]]; in executeOperation()
1467 const RunTimeOperandInfo& bias = operands[ins[2]]; in executeOperation()
1496 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1499 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1601 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1602 const RunTimeOperandInfo& multiples = operands[ins[1]]; in executeOperation()
1604 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1620 RunTimeOperandInfo& cellStateOut = in executeOperation()
1622 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation()
1637 const RunTimeOperandInfo& base = operands[ins[0]]; in executeOperation()
1638 const RunTimeOperandInfo& exponent = operands[ins[1]]; in executeOperation()
1640 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1685 static void setInfoExceptLifetime(RunTimeOperandInfo* to, const RunTimeOperandInfo& from) { in setInfoExceptLifetime()
1693 int CpuExecutor::executeIfOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeIfOperation()
1695 const RunTimeOperandInfo& condOperand = operands[operation.inputs[op::kCondBoolOperand]]; in executeIfOperation()
1704 const RunTimeOperandInfo& branchOperand = operands[operation.inputs[branchInputIndex]]; in executeIfOperation()
1707 std::vector<RunTimeOperandInfo> branchOperands = initializeRunTimeInfo(branchSubgraph); in executeIfOperation()
1732 int CpuExecutor::executeWhileOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeWhileOperation()
1734 const RunTimeOperandInfo& condModelOperand = operands[operation.inputs[op::kCondModelOperand]]; in executeWhileOperation()
1735 const RunTimeOperandInfo& bodyModelOperand = operands[operation.inputs[op::kBodyModelOperand]]; in executeWhileOperation()
1740 std::vector<RunTimeOperandInfo> condOperands = initializeRunTimeInfo(condSubgraph); in executeWhileOperation()
1741 std::vector<RunTimeOperandInfo> bodyOperands = initializeRunTimeInfo(bodySubgraph); in executeWhileOperation()
1788 RunTimeOperandInfo& condOutput = condOperands[condSubgraph.outputIndexes[0]]; in executeWhileOperation()
1826 RunTimeOperandInfo& info = bodyOperands[bodySubgraph.outputIndexes[i]]; in executeWhileOperation()
1848 RunTimeOperandInfo& outerOperand = operands[operation.outputs[i]]; in executeWhileOperation()
1849 RunTimeOperandInfo& innerOperand = condOperands[condSubgraph.inputIndexes[i]]; in executeWhileOperation()
1862 const std::vector<RunTimeOperandInfo>& operands) { in setOutputShapes()
1866 const RunTimeOperandInfo& from = operands[operandIndex]; in setOutputShapes()