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()
718 int CpuExecutor::executeOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeOperation()
798 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
799 const RunTimeOperandInfo& targetShape = operands[ins[1]]; in executeOperation()
801 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
815 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
819 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
822 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
879 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
883 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
886 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
943 const RunTimeOperandInfo& values = operands[ins[EmbeddingLookup::kValueTensor]]; in executeOperation()
944 const RunTimeOperandInfo& lookups = operands[ins[EmbeddingLookup::kLookupTensor]]; in executeOperation()
945 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()
957 const RunTimeOperandInfo& lookups = operands[ins[HashtableLookup::kLookupTensor]]; in executeOperation()
958 const RunTimeOperandInfo& keys = operands[ins[HashtableLookup::kKeyTensor]]; in executeOperation()
959 const RunTimeOperandInfo& values = operands[ins[HashtableLookup::kValueTensor]]; in executeOperation()
961 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation()
962 RunTimeOperandInfo& hits = operands[outs[HashtableLookup::kHitsTensor]]; in executeOperation()
973 RunTimeOperandInfo& output = operands[outs[LSHProjection::kOutputTensor]]; in executeOperation()
981 const RunTimeOperandInfo& hash = operands[ins[LSHProjection::kHashTensor]]; in executeOperation()
1001 RunTimeOperandInfo& fwOutput = in executeOperation()
1012 RunTimeOperandInfo& bwOutput = in executeOperation()
1018 RunTimeOperandInfo& fwOutputActivationState = in executeOperation()
1021 RunTimeOperandInfo& fwOutputCellState = in executeOperation()
1023 RunTimeOperandInfo& bwOutputActivationState = in executeOperation()
1026 RunTimeOperandInfo& bwOutputCellState = in executeOperation()
1041 RunTimeOperandInfo& scratch = operands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1042 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1043 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1044 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1060 RunTimeOperandInfo& output = operands[outs[Multinomial::kOutputTensor]]; in executeOperation()
1074 RunTimeOperandInfo& hiddenStateOut = operands[outs[RNN::kHiddenStateOutTensor]]; in executeOperation()
1075 RunTimeOperandInfo& output = operands[outs[RNN::kOutputTensor]]; in executeOperation()
1085 RunTimeOperandInfo& stateOut = operands[outs[SVDF::kStateOutTensor]]; in executeOperation()
1086 RunTimeOperandInfo& output = operands[outs[SVDF::kOutputTensor]]; in executeOperation()
1100 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1101 const RunTimeOperandInfo& blockSize = operands[ins[1]]; in executeOperation()
1104 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1107 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1171 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1172 const RunTimeOperandInfo& blockSize = operands[ins[1]]; in executeOperation()
1173 const RunTimeOperandInfo& paddings = operands[ins[2]]; in executeOperation()
1176 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1179 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1249 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1250 const RunTimeOperandInfo& paddings = operands[ins[1]]; in executeOperation()
1252 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1289 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1291 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1302 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1303 const RunTimeOperandInfo& axis = operands[ins[1]]; in executeOperation()
1306 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1341 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1344 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1357 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1360 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1373 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1447 const RunTimeOperandInfo& in1 = operands[ins[0]]; in executeOperation()
1448 const RunTimeOperandInfo& in2 = operands[ins[1]]; in executeOperation()
1450 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1464 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1465 const RunTimeOperandInfo& filter = operands[ins[1]]; in executeOperation()
1466 const RunTimeOperandInfo& bias = operands[ins[2]]; in executeOperation()
1495 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1498 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1600 const RunTimeOperandInfo& input = operands[ins[0]]; in executeOperation()
1601 const RunTimeOperandInfo& multiples = operands[ins[1]]; in executeOperation()
1603 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1619 RunTimeOperandInfo& cellStateOut = in executeOperation()
1621 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation()
1636 const RunTimeOperandInfo& base = operands[ins[0]]; in executeOperation()
1637 const RunTimeOperandInfo& exponent = operands[ins[1]]; in executeOperation()
1639 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1684 static void setInfoExceptLifetime(RunTimeOperandInfo* to, const RunTimeOperandInfo& from) { in setInfoExceptLifetime()
1692 int CpuExecutor::executeIfOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeIfOperation()
1694 const RunTimeOperandInfo& condOperand = operands[operation.inputs[op::kCondBoolOperand]]; in executeIfOperation()
1703 const RunTimeOperandInfo& branchOperand = operands[operation.inputs[branchInputIndex]]; in executeIfOperation()
1706 std::vector<RunTimeOperandInfo> branchOperands = initializeRunTimeInfo(branchSubgraph); in executeIfOperation()
1731 int CpuExecutor::executeWhileOperation(const Operation& operation, RunTimeOperandInfo* operands) { in executeWhileOperation()
1733 const RunTimeOperandInfo& condModelOperand = operands[operation.inputs[op::kCondModelOperand]]; in executeWhileOperation()
1734 const RunTimeOperandInfo& bodyModelOperand = operands[operation.inputs[op::kBodyModelOperand]]; in executeWhileOperation()
1739 std::vector<RunTimeOperandInfo> condOperands = initializeRunTimeInfo(condSubgraph); in executeWhileOperation()
1740 std::vector<RunTimeOperandInfo> bodyOperands = initializeRunTimeInfo(bodySubgraph); in executeWhileOperation()
1787 RunTimeOperandInfo& condOutput = condOperands[condSubgraph.outputIndexes[0]]; in executeWhileOperation()
1825 RunTimeOperandInfo& info = bodyOperands[bodySubgraph.outputIndexes[i]]; in executeWhileOperation()
1847 RunTimeOperandInfo& outerOperand = operands[operation.outputs[i]]; in executeWhileOperation()
1848 RunTimeOperandInfo& innerOperand = condOperands[condSubgraph.inputIndexes[i]]; in executeWhileOperation()
1861 const std::vector<RunTimeOperandInfo>& operands) { in setOutputShapes()
1865 const RunTimeOperandInfo& from = operands[operandIndex]; in setOutputShapes()