Cross Reference: /frameworks/ml/nn/common/operations/LSTM.cpp

Lines Matching refs:operation
39 LSTMCell::LSTMCell(const Operation& operation,
41   input_ = GetInput(operation, operands, kInputTensor);
43   input_to_input_weights_ = GetInput(operation, operands, kInputToInputWeightsTensor);  // optional
44   input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor);
45   input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor);
46   input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor);
49       GetInput(operation, operands, kRecurrentToInputWeightsTensor);  // optional
50   recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor);
51   recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor);
52   recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor);
54   cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor);    // optional
55   cell_to_forget_weights_ = GetInput(operation, operands, kCellToForgetWeightsTensor);  // optional
56   cell_to_output_weights_ = GetInput(operation, operands, kCellToOutputWeightsTensor);  // optional
58   input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor);
59   forget_gate_bias_ = GetInput(operation, operands, kForgetGateBiasTensor);
60   cell_bias_ = GetInput(operation, operands, kCellGateBiasTensor);
61   output_gate_bias_ = GetInput(operation, operands, kOutputGateBiasTensor);
63   projection_weights_ = GetInput(operation, operands, kProjectionWeightsTensor);  // optional
64   projection_bias_ = GetInput(operation, operands, kProjectionBiasTensor);        // optional
66   output_state_in_ = GetInput(operation, operands, kOutputStateInTensor);
67   cell_state_in_ = GetInput(operation, operands, kCellStateInTensor);
70       *GetInput(operation, operands, kActivationParam)));
71   params_.cell_clip_ = getScalarData<float>(*GetInput(operation, operands, kCellClipParam));
72   params_.proj_clip_ = getScalarData<float>(*GetInput(operation, operands, kProjClipParam));
74   output_state_out_ = GetOutput(operation, operands, kOutputStateOutTensor);
75   cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor);
76   output_ = GetOutput(operation, operands, kOutputTensor);
78   scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor);
82     const Operation &operation, std::vector<RunTimeOperandInfo> &operands,
86         *GetInput(operation, operands, LSTMCell::kActivationParam))),
87     .cell_clip_  = getScalarData<float>(*GetInput(operation, operands, LSTMCell::kCellClipParam)),
88     .proj_clip_  = getScalarData<float>(*GetInput(operation, operands, LSTMCell::kProjClipParam))
98       GetInput(operation, operands, LSTMCell::kInputToInputWeightsTensor);
106       GetInput(operation, operands, LSTMCell::kInputToForgetWeightsTensor);
112       GetInput(operation, operands, LSTMCell::kInputToCellWeightsTensor);
118       GetInput(operation, operands, LSTMCell::kRecurrentToInputWeightsTensor);
126       GetInput(operation, operands, LSTMCell::kRecurrentToForgetWeightsTensor);
132       GetInput(operation, operands, LSTMCell::kRecurrentToCellWeightsTensor);
147       GetInput(operation, operands, LSTMCell::kCellToInputWeightsTensor);
154       GetInput(operation, operands, LSTMCell::kCellToForgetWeightsTensor);
161       GetInput(operation, operands, LSTMCell::kCellToOutputWeightsTensor);
180       GetInput(operation, operands, LSTMCell::kInputGateBiasTensor);
189       GetInput(operation, operands, LSTMCell::kForgetGateBiasTensor);
194       GetInput(operation, operands, LSTMCell::kCellGateBiasTensor);
199       GetInput(operation, operands, LSTMCell::kOutputGateBiasTensor);
204       GetInput(operation, operands, LSTMCell::kProjectionWeightsTensor);
212       GetInput(operation, operands, LSTMCell::kProjectionBiasTensor);
230 bool LSTMCell::Prepare(const Operation &operation,
237   NN_CHECK(NumInputsWithValues(operation, operands) >= 15 &&
238            NumInputsWithValues(operation, operands) <= 23);
239   NN_CHECK_EQ(NumOutputs(operation), 4);
244       GetInput(operation, operands, LSTMCell::kInputTensor);
250       GetInput(operation, operands, LSTMCell::kInputToOutputWeightsTensor);
256       GetInput(operation, operands, LSTMCell::kRecurrentToOutputWeightsTensor);
263   if (!CheckInputTensorDimensions(operation, operands, n_input, n_output, n_cell)) {
286       GetInput(operation, operands, LSTMCell::kInputToInputWeightsTensor);