Lines Matching refs:operands

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