Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp83 input_ = GetInput(operation, operands, kInputTensor); in BidirectionalSequenceLSTM()
86 GetInput(operation, operands, kFwInputToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
87 fw_input_to_forget_weights_ = GetInput(operation, operands, kFwInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
88 fw_input_to_cell_weights_ = GetInput(operation, operands, kFwInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
89 fw_input_to_output_weights_ = GetInput(operation, operands, kFwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
92 GetInput(operation, operands, kFwRecurrentToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
94 GetInput(operation, operands, kFwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
95 fw_recurrent_to_cell_weights_ = GetInput(operation, operands, kFwRecurrentToCellWeightsTensor); in BidirectionalSequenceLSTM()
97 GetInput(operation, operands, kFwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
100 GetInput(operation, operands, kFwCellToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
[all …]
DQuantizedLSTM.cpp224 input_ = GetInput(operation, operands, kInputTensor); in QuantizedLSTMCell()
226 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell()
227 inputToForgetWeights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in QuantizedLSTMCell()
228 inputToCellWeights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in QuantizedLSTMCell()
229 inputToOutputWeights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in QuantizedLSTMCell()
231 recurrentToInputWeights_ = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in QuantizedLSTMCell()
232 recurrentToForgetWeights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in QuantizedLSTMCell()
233 recurrentToCellWeights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in QuantizedLSTMCell()
234 recurrentToOutputWeights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in QuantizedLSTMCell()
236 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 …]
DLSHProjection.cpp35 input_ = GetInput(operation, operands, kInputTensor); in LSHProjection()
36 weight_ = GetInput(operation, operands, kWeightTensor); in LSHProjection()
37 hash_ = GetInput(operation, operands, kHashTensor); in LSHProjection()
40 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()
50 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
55 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
60 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
73 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
DLSTM.cpp55 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
58 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
59 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
60 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()
61 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
64 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
65 recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in LSTMCell()
66 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
67 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
69 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
[all …]
DRNN.cpp36 input_ = GetInput(operation, operands, kInputTensor); in RNN()
37 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
38 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()
39 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor); in RNN()
40 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
57 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
58 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare()
60 GetInput(operation, operands, kRecurrentWeightsTensor); in Prepare()
61 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
DMultinomial.cpp56 input_ = GetInput(operation, operands, kInputTensor); in Multinomial()
57 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()
58 random_seeds_ = GetInput(operation, operands, kRandomSeedsTensor); in Multinomial()
69 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
74 getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Prepare()
DHashtableLookup.cpp41 lookup_ = GetInput(operation, operands, kLookupTensor); in HashtableLookup()
42 key_ = GetInput(operation, operands, kKeyTensor); in HashtableLookup()
43 value_ = GetInput(operation, operands, kValueTensor); in HashtableLookup()
DEmbeddingLookup.cpp33 value_ = GetInput(operation, operands, kValueTensor); in EmbeddingLookup()
34 lookup_ = GetInput(operation, operands, kLookupTensor); in EmbeddingLookup()
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()
/frameworks/ml/nn/common/include/
DCpuExecutor.h284 inline RunTimeOperandInfo* GetInput(const hal::Operation& operation, RunTimeOperandInfo* operands, in GetInput() function