Lines Matching refs:operands
82 RunTimeOperandInfo* operands) { in BidirectionalSequenceLSTM() argument
83 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()
102 GetInput(operation, operands, kFwCellToForgetWeightsTensor); // optional in BidirectionalSequenceLSTM()
104 GetInput(operation, operands, kFwCellToOutputWeightsTensor); // optional in BidirectionalSequenceLSTM()
106 fw_input_gate_bias_ = GetInput(operation, operands, kFwInputGateBiasTensor); in BidirectionalSequenceLSTM()
107 fw_forget_gate_bias_ = GetInput(operation, operands, kFwForgetGateBiasTensor); in BidirectionalSequenceLSTM()
108 fw_cell_bias_ = GetInput(operation, operands, kFwCellGateBiasTensor); in BidirectionalSequenceLSTM()
109 fw_output_gate_bias_ = GetInput(operation, operands, kFwOutputGateBiasTensor); in BidirectionalSequenceLSTM()
111 fw_projection_weights_ = GetInput(operation, operands, kFwProjectionWeightsTensor); // optional in BidirectionalSequenceLSTM()
112 fw_projection_bias_ = GetInput(operation, operands, kFwProjectionBiasTensor); // optional in BidirectionalSequenceLSTM()
114 fw_activation_state_ = GetInput(operation, operands, kFwInputActivationStateTensor); in BidirectionalSequenceLSTM()
115 fw_cell_state_ = GetInput(operation, operands, kFwInputCellStateTensor); in BidirectionalSequenceLSTM()
118 GetInput(operation, operands, kBwInputToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
119 bw_input_to_forget_weights_ = GetInput(operation, operands, kBwInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
120 bw_input_to_cell_weights_ = GetInput(operation, operands, kBwInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
121 bw_input_to_output_weights_ = GetInput(operation, operands, kBwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
124 GetInput(operation, operands, kBwRecurrentToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
126 GetInput(operation, operands, kBwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
127 bw_recurrent_to_cell_weights_ = GetInput(operation, operands, kBwRecurrentToCellWeightsTensor); in BidirectionalSequenceLSTM()
129 GetInput(operation, operands, kBwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
132 GetInput(operation, operands, kBwCellToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
134 GetInput(operation, operands, kBwCellToForgetWeightsTensor); // optional in BidirectionalSequenceLSTM()
136 GetInput(operation, operands, kBwCellToOutputWeightsTensor); // optional in BidirectionalSequenceLSTM()
138 bw_input_gate_bias_ = GetInput(operation, operands, kBwInputGateBiasTensor); in BidirectionalSequenceLSTM()
139 bw_forget_gate_bias_ = GetInput(operation, operands, kBwForgetGateBiasTensor); in BidirectionalSequenceLSTM()
140 bw_cell_bias_ = GetInput(operation, operands, kBwCellGateBiasTensor); in BidirectionalSequenceLSTM()
141 bw_output_gate_bias_ = GetInput(operation, operands, kBwOutputGateBiasTensor); in BidirectionalSequenceLSTM()
143 bw_projection_weights_ = GetInput(operation, operands, kBwProjectionWeightsTensor); // optional in BidirectionalSequenceLSTM()
144 bw_projection_bias_ = GetInput(operation, operands, kBwProjectionBiasTensor); // optional in BidirectionalSequenceLSTM()
146 bw_activation_state_ = GetInput(operation, operands, kBwInputActivationStateTensor); in BidirectionalSequenceLSTM()
147 bw_cell_state_ = GetInput(operation, operands, kBwInputCellStateTensor); in BidirectionalSequenceLSTM()
149 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()
150 fw_aux_input_to_input_weights_ = GetInput(operation, operands, kFwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()
152 GetInput(operation, operands, kFwAuxInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
153 fw_aux_input_to_cell_weights_ = GetInput(operation, operands, kFwAuxInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
155 GetInput(operation, operands, kFwAuxInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
156 bw_aux_input_to_input_weights_ = GetInput(operation, operands, kBwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()
158 GetInput(operation, operands, kBwAuxInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
159 bw_aux_input_to_cell_weights_ = GetInput(operation, operands, kBwAuxInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
161 GetInput(operation, operands, kBwAuxInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
163 fw_input_layer_norm_weights_ = GetInput(operation, operands, kFwInputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
164 fw_forget_layer_norm_weights_ = GetInput(operation, operands, kFwForgetLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
165 fw_cell_layer_norm_weights_ = GetInput(operation, operands, kFwCellLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
166 fw_output_layer_norm_weights_ = GetInput(operation, operands, kFwOutputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
167 bw_input_layer_norm_weights_ = GetInput(operation, operands, kBwInputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
168 bw_forget_layer_norm_weights_ = GetInput(operation, operands, kBwForgetLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
169 bw_cell_layer_norm_weights_ = GetInput(operation, operands, kBwCellLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
170 bw_output_layer_norm_weights_ = GetInput(operation, operands, kBwOutputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
172 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in BidirectionalSequenceLSTM()
175 const auto& clipOperand = *GetInput(operation, operands, kCellClipParam); in BidirectionalSequenceLSTM()
176 const auto& projOperand = *GetInput(operation, operands, kProjClipParam); in BidirectionalSequenceLSTM()
186 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in BidirectionalSequenceLSTM()
188 const auto& timeMajorOperand = *GetInput(operation, operands, kTimeMajorParam); in BidirectionalSequenceLSTM()
192 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
194 bw_output_ = GetOutput(operation, operands, kBwOutputTensor); in BidirectionalSequenceLSTM()
201 GetOutput(operation, operands, kFwOutputActivationStateTensor - delta); in BidirectionalSequenceLSTM()
202 fw_output_cell_state_ = GetOutput(operation, operands, kFwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
204 GetOutput(operation, operands, kBwOutputActivationStateTensor - delta); in BidirectionalSequenceLSTM()
205 bw_output_cell_state_ = GetOutput(operation, operands, kBwOutputCellStateTensor - delta); in BidirectionalSequenceLSTM()
209 bool BidirectionalSequenceLSTM::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare() argument
245 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
250 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
252 const auto& cellOperand = *GetInput(operation, operands, kCellClipParam); in Prepare()
253 const auto& projOperand = *GetInput(operation, operands, kProjClipParam); in Prepare()
261 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in Prepare()
263 const auto& timeMajorOperand = *GetInput(operation, operands, kTimeMajorParam); in Prepare()