Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp50 input_ = GetInput(operation, operands, kInputTensor); in BidirectionalSequenceLSTM()
53 GetInput(operation, operands, kFwInputToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
54 fw_input_to_forget_weights_ = GetInput(operation, operands, kFwInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
55 fw_input_to_cell_weights_ = GetInput(operation, operands, kFwInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
56 fw_input_to_output_weights_ = GetInput(operation, operands, kFwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
59 GetInput(operation, operands, kFwRecurrentToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
61 GetInput(operation, operands, kFwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
62 fw_recurrent_to_cell_weights_ = GetInput(operation, operands, kFwRecurrentToCellWeightsTensor); in BidirectionalSequenceLSTM()
64 GetInput(operation, operands, kFwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
67 GetInput(operation, operands, kFwCellToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
[all …]
DQuantizedLSTM.cpp218 input_ = GetInput(operation, operands, kInputTensor); in QuantizedLSTMCell()
220 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell()
221 inputToForgetWeights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in QuantizedLSTMCell()
222 inputToCellWeights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in QuantizedLSTMCell()
223 inputToOutputWeights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in QuantizedLSTMCell()
225 recurrentToInputWeights_ = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in QuantizedLSTMCell()
226 recurrentToForgetWeights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in QuantizedLSTMCell()
227 recurrentToCellWeights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in QuantizedLSTMCell()
228 recurrentToOutputWeights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in QuantizedLSTMCell()
230 inputGateBias_ = GetInput(operation, operands, kInputGateBiasTensor); in QuantizedLSTMCell()
[all …]
DSVDF.cpp31 input_ = GetInput(operation, operands, kInputTensor); in SVDF()
32 weights_feature_ = GetInput(operation, operands, kWeightsFeatureTensor); in SVDF()
33 weights_time_ = GetInput(operation, operands, kWeightsTimeTensor); in SVDF()
34 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
35 state_in_ = GetInput(operation, operands, kStateInTensor); in SVDF()
37 params_.rank_ = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in SVDF()
39 *GetInput(operation, operands, kActivationParam))); in SVDF()
57 GetInput(operation, operands, SVDF::kInputTensor); in Prepare()
59 GetInput(operation, operands, SVDF::kWeightsFeatureTensor); in Prepare()
61 GetInput(operation, operands, SVDF::kWeightsTimeTensor); in Prepare()
[all …]
DRNN.cpp31 input_ = GetInput(operation, operands, kInputTensor); in RNN()
32 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
33 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()
34 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor); in RNN()
35 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
55 GetInput(operation, operands, kInputTensor); in Prepare()
57 GetInput(operation, operands, kWeightsTensor); in Prepare()
59 GetInput(operation, operands, kRecurrentWeightsTensor); in Prepare()
61 GetInput(operation, operands, kBiasTensor); in Prepare()
DLSHProjection.cpp30 input_ = GetInput(operation, operands, kInputTensor); in LSHProjection()
31 weight_ = GetInput(operation, operands, kWeightTensor); in LSHProjection()
32 hash_ = GetInput(operation, operands, kHashTensor); in LSHProjection()
35 getScalarData<int32_t>(*GetInput(operation, operands, kTypeParam))); in LSHProjection()
46 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
51 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
63 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
DLSTM.cpp50 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
53 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
54 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
55 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()
56 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
59 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
60 recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in LSTMCell()
61 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
62 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
64 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
[all …]
DMultinomial.cpp49 input_ = GetInput(operation, operands, kInputTensor); in Multinomial()
50 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()
51 random_seeds_ = GetInput(operation, operands, kRandomSeedsTensor); in Multinomial()
62 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
67 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.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.h261 inline RunTimeOperandInfo *GetInput(const Operation &operation, in GetInput() function