Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp82 input_ = GetInput(operation, operands, kInputTensor); in BidirectionalSequenceLSTM()
85 GetInput(operation, operands, kFwInputToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
86 fw_input_to_forget_weights_ = GetInput(operation, operands, kFwInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
87 fw_input_to_cell_weights_ = GetInput(operation, operands, kFwInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
88 fw_input_to_output_weights_ = GetInput(operation, operands, kFwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
91 GetInput(operation, operands, kFwRecurrentToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
93 GetInput(operation, operands, kFwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
94 fw_recurrent_to_cell_weights_ = GetInput(operation, operands, kFwRecurrentToCellWeightsTensor); in BidirectionalSequenceLSTM()
96 GetInput(operation, operands, kFwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
99 GetInput(operation, operands, kFwCellToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
[all …]
DQuantizedLSTM.cpp225 input_ = GetInput(operation, operands, kInputTensor); in QuantizedLSTMCell()
227 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell()
228 inputToForgetWeights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in QuantizedLSTMCell()
229 inputToCellWeights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in QuantizedLSTMCell()
230 inputToOutputWeights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in QuantizedLSTMCell()
232 recurrentToInputWeights_ = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in QuantizedLSTMCell()
233 recurrentToForgetWeights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in QuantizedLSTMCell()
234 recurrentToCellWeights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in QuantizedLSTMCell()
235 recurrentToOutputWeights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in QuantizedLSTMCell()
237 inputGateBias_ = GetInput(operation, operands, kInputGateBiasTensor); in QuantizedLSTMCell()
[all …]
DSVDF.cpp35 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
36 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
37 weights_time_ = GetInput(operation, operands, kWeightsTimeTensor); in SVDF()
38 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
39 state_in_ = GetInput(operation, operands, kStateInTensor); in SVDF()
41 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in SVDF()
43 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in SVDF()
63 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
69 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in Prepare()
71 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
[all …]
DLSTM.cpp60 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
63 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
64 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
65 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()
66 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
69 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
70 recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in LSTMCell()
71 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
72 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
74 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
[all …]
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()
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()
DMultinomial.cpp62 input_ = GetInput(operation, operands, kInputTensor); in Multinomial()
63 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()
64 random_seeds_ = GetInput(operation, operands, kRandomSeedsTensor); in Multinomial()
75 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
80 getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Prepare()
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()
DEmbeddingLookup.cpp29 value_ = GetInput(operation, operands, kValueTensor); in EmbeddingLookup()
30 lookup_ = GetInput(operation, operands, kLookupTensor); in EmbeddingLookup()
DMultinomialTest.cpp96 const std::vector<float>& GetInput() const { return input_; } in GetInput() function in android::nn::wrapper::MultinomialOpModel
126 std::vector<float> input = multinomial.GetInput(); in TEST()
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h283 inline RunTimeOperandInfo* GetInput(const Operation& operation, RunTimeOperandInfo* operands, in GetInput() function