Home
last modified time | relevance | path

Searched refs:GetInput (Results 1 – 20 of 20) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceLSTM.cpp82 input_ = GetInput(operation, operands, kInputTensor); in BidirectionalSequenceLSTM()
93 GetInput(operation, operands, kFwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
96 GetInput(operation, operands, kFwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
107 fw_cell_bias_ = GetInput(operation, operands, kFwCellGateBiasTensor); in BidirectionalSequenceLSTM()
114 fw_cell_state_ = GetInput(operation, operands, kFwInputCellStateTensor); in BidirectionalSequenceLSTM()
139 bw_cell_bias_ = GetInput(operation, operands, kBwCellGateBiasTensor); in BidirectionalSequenceLSTM()
146 bw_cell_state_ = GetInput(operation, operands, kBwInputCellStateTensor); in BidirectionalSequenceLSTM()
148 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()
151 GetInput(operation, operands, kFwAuxInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
154 GetInput(operation, operands, kFwAuxInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
[all …]
H A DQuantizedLSTM.cpp221 input_ = GetInput(operation, operands, kInputTensor); in QuantizedLSTMCell()
233 inputGateBias_ = GetInput(operation, operands, kInputGateBiasTensor); in QuantizedLSTMCell()
234 forgetGateBias_ = GetInput(operation, operands, kForgetGateBiasTensor); in QuantizedLSTMCell()
235 cellGateBias_ = GetInput(operation, operands, kCellGateBiasTensor); in QuantizedLSTMCell()
236 outputGateBias_ = GetInput(operation, operands, kOutputGateBiasTensor); in QuantizedLSTMCell()
238 prevCellState_ = GetInput(operation, operands, kPrevCellStateTensor); in QuantizedLSTMCell()
239 prevOutput_ = GetInput(operation, operands, kPrevOutputTensor); in QuantizedLSTMCell()
247 auto input = GetInput(operation, operands, kInputTensor); in prepare()
254 auto prevOutput = GetInput(operation, operands, kPrevOutputTensor); in prepare()
292 auto inputGateBias = GetInput(operation, operands, kInputGateBiasTensor); in prepare()
[all …]
H A DSVDF.cpp33 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
34 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
35 weights_time_ = GetInput(operation, operands, kWeightsTimeTensor); in SVDF()
36 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
37 state_in_ = GetInput(operation, operands, kStateInTensor); in SVDF()
39 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in SVDF()
61 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
67 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in Prepare()
74 GetInput(operation, operands, SVDF::kWeightsFeatureTensor); in Prepare()
76 GetInput(operation, operands, SVDF::kWeightsTimeTensor); in Prepare()
[all …]
H A DLSHProjection.cpp34 input_ = GetInput(operation, operands, kInputTensor); in LSHProjection()
35 weight_ = GetInput(operation, operands, kWeightTensor); in LSHProjection()
36 hash_ = GetInput(operation, operands, kHashTensor); in LSHProjection()
39 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()
49 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
54 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
59 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
72 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
H A DLSTM.cpp55 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
58 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
71 GetInput(operation, operands, kCellToForgetWeightsTensor); // optional in LSTMCell()
75 input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor); in LSTMCell()
76 forget_gate_bias_ = GetInput(operation, operands, kForgetGateBiasTensor); in LSTMCell()
77 cell_bias_ = GetInput(operation, operands, kCellGateBiasTensor); in LSTMCell()
78 output_gate_bias_ = GetInput(operation, operands, kOutputGateBiasTensor); in LSTMCell()
83 output_state_in_ = GetInput(operation, operands, kOutputStateInTensor); in LSTMCell()
84 cell_state_in_ = GetInput(operation, operands, kCellStateInTensor); in LSTMCell()
90 const auto& cellClipOperand = *GetInput(operation, operands, kCellClipParam); in LSTMCell()
[all …]
H A DRNN.cpp32 input_ = GetInput(operation, operands, kInputTensor); in RNN()
33 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
34 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()
35 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor); in RNN()
36 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
53 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
54 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare()
56 GetInput(operation, operands, kRecurrentWeightsTensor); in Prepare()
57 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
H A DMultinomial.cpp58 input_ = GetInput(operation, operands, kInputTensor); in Multinomial()
59 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()
60 random_seeds_ = GetInput(operation, operands, kRandomSeedsTensor); in Multinomial()
71 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
76 getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Prepare()
H A DHashtableLookup.cpp37 lookup_ = GetInput(operation, operands, kLookupTensor); in HashtableLookup()
38 key_ = GetInput(operation, operands, kKeyTensor); in HashtableLookup()
39 value_ = GetInput(operation, operands, kValueTensor); in HashtableLookup()
H A DEmbeddingLookup.cpp29 value_ = GetInput(operation, operands, kValueTensor); in EmbeddingLookup()
30 lookup_ = GetInput(operation, operands, kLookupTensor); in EmbeddingLookup()
H A DMultinomialTest.cpp91 const std::vector<float>& GetInput() const { return input_; } in GetInput() function in android::nn::wrapper::MultinomialOpModel
121 std::vector<float> input = multinomial.GetInput(); in TEST()
/aosp12/art/compiler/optimizing/
H A Dnodes.cc1770 if (GetInput()->IsIntConstant()) { in TryStaticEvaluation()
1790 } else if (GetInput()->IsLongConstant()) { in TryStaticEvaluation()
1810 } else if (GetInput()->IsFloatConstant()) { in TryStaticEvaluation()
1834 } else if (GetInput()->IsDoubleConstant()) { in TryStaticEvaluation()
1863 if (GetInput()->IsIntConstant()) { in TryStaticEvaluation()
1864 return Evaluate(GetInput()->AsIntConstant()); in TryStaticEvaluation()
1865 } else if (GetInput()->IsLongConstant()) { in TryStaticEvaluation()
1866 return Evaluate(GetInput()->AsLongConstant()); in TryStaticEvaluation()
1868 if (GetInput()->IsFloatConstant()) { in TryStaticEvaluation()
1869 return Evaluate(GetInput()->AsFloatConstant()); in TryStaticEvaluation()
[all …]
H A Dinstruction_simplifier.cc210 binop->ReplaceInput(left_neg->GetInput(), 0); in TryMoveNegOnInputsAfterBinop()
211 binop->ReplaceInput(right_neg->GetInput(), 1); in TryMoveNegOnInputsAfterBinop()
423 instruction->ReplaceInput(shift_amount->AsTypeConversion()->GetInput(), 1); in VisitShift()
1257 HInstruction* input = instruction->GetInput(); in VisitTypeConversion()
1269 HInstruction* original_input = input_conversion->GetInput(); in VisitTypeConversion()
1353 HInstruction* input = instruction->GetInput(); in VisitAbs()
1407 new(GetGraph()->GetAllocator()) HSub(instruction->GetType(), other, neg->GetInput()); in VisitAdd()
1974 HInstruction* input = instruction->GetInput(); in VisitNeg()
1982 instruction->ReplaceWith(previous_neg->GetInput()); in VisitNeg()
2019 HInstruction* input = instruction->GetInput(); in VisitNot()
[all …]
H A Dinstruction_simplifier_x86_shared.cc45 not_ins->GetInput(), in TryCombineAndNot()
H A Dprepare_for_register_allocation.cc314 instruction->ReplaceWith(instruction->GetInput()); in VisitTypeConversion()
H A Dinstruction_simplifier_shared.cc217 HInstruction* src = hnot->AsNot()->GetInput(); in TryMergeNegatedInput()
H A Dconstant_folding.cc242 HInstruction* src = hnot->AsNot()->GetInput(); in VisitAnd()
H A Dnodes_vector.h284 HInstruction* GetInput() const { return InputAt(0); } in GetInput() function
H A Dinduction_var_analysis.cc878 auto it = cycle_.find(conversion->GetInput()); in SolveConversion()
H A Dnodes.h3743 HInstruction* GetInput() const { return InputAt(0); }
6033 HInstruction* GetInput() const { return InputAt(0); }
6034 DataType::Type GetInputType() const { return GetInput()->GetType(); }
/aosp12/packages/modules/NeuralNetworks/common/include/
H A DCpuExecutor.h283 inline RunTimeOperandInfo* GetInput(const Operation& operation, RunTimeOperandInfo* operands, in GetInput() function