Lines Matching refs:RunTimeOperandInfo
31 struct RunTimeOperandInfo;
35 BidirectionalSequenceLSTM(const Operation& operation, RunTimeOperandInfo* operands);
37 bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* fwOutputShape,
160 const RunTimeOperandInfo* input_;
162 const RunTimeOperandInfo* aux_input_;
163 const RunTimeOperandInfo* fw_aux_input_to_input_weights_;
164 const RunTimeOperandInfo* fw_aux_input_to_forget_weights_;
165 const RunTimeOperandInfo* fw_aux_input_to_cell_weights_;
166 const RunTimeOperandInfo* fw_aux_input_to_output_weights_;
167 const RunTimeOperandInfo* bw_aux_input_to_input_weights_;
168 const RunTimeOperandInfo* bw_aux_input_to_forget_weights_;
169 const RunTimeOperandInfo* bw_aux_input_to_cell_weights_;
170 const RunTimeOperandInfo* bw_aux_input_to_output_weights_;
172 const RunTimeOperandInfo* fw_input_to_input_weights_;
173 const RunTimeOperandInfo* fw_input_to_forget_weights_;
174 const RunTimeOperandInfo* fw_input_to_cell_weights_;
175 const RunTimeOperandInfo* fw_input_to_output_weights_;
177 const RunTimeOperandInfo* fw_recurrent_to_input_weights_;
178 const RunTimeOperandInfo* fw_recurrent_to_forget_weights_;
179 const RunTimeOperandInfo* fw_recurrent_to_cell_weights_;
180 const RunTimeOperandInfo* fw_recurrent_to_output_weights_;
182 const RunTimeOperandInfo* fw_cell_to_input_weights_;
183 const RunTimeOperandInfo* fw_cell_to_forget_weights_;
184 const RunTimeOperandInfo* fw_cell_to_output_weights_;
186 const RunTimeOperandInfo* fw_input_gate_bias_;
187 const RunTimeOperandInfo* fw_forget_gate_bias_;
188 const RunTimeOperandInfo* fw_cell_bias_;
189 const RunTimeOperandInfo* fw_output_gate_bias_;
191 const RunTimeOperandInfo* fw_projection_weights_;
192 const RunTimeOperandInfo* fw_projection_bias_;
194 const RunTimeOperandInfo* fw_input_layer_norm_weights_;
195 const RunTimeOperandInfo* fw_forget_layer_norm_weights_;
196 const RunTimeOperandInfo* fw_cell_layer_norm_weights_;
197 const RunTimeOperandInfo* fw_output_layer_norm_weights_;
199 const RunTimeOperandInfo* fw_activation_state_;
200 const RunTimeOperandInfo* fw_cell_state_;
201 RunTimeOperandInfo* fw_output_;
203 const RunTimeOperandInfo* bw_input_to_input_weights_;
204 const RunTimeOperandInfo* bw_input_to_forget_weights_;
205 const RunTimeOperandInfo* bw_input_to_cell_weights_;
206 const RunTimeOperandInfo* bw_input_to_output_weights_;
208 const RunTimeOperandInfo* bw_recurrent_to_input_weights_;
209 const RunTimeOperandInfo* bw_recurrent_to_forget_weights_;
210 const RunTimeOperandInfo* bw_recurrent_to_cell_weights_;
211 const RunTimeOperandInfo* bw_recurrent_to_output_weights_;
213 const RunTimeOperandInfo* bw_cell_to_input_weights_;
214 const RunTimeOperandInfo* bw_cell_to_forget_weights_;
215 const RunTimeOperandInfo* bw_cell_to_output_weights_;
217 const RunTimeOperandInfo* bw_input_gate_bias_;
218 const RunTimeOperandInfo* bw_forget_gate_bias_;
219 const RunTimeOperandInfo* bw_cell_bias_;
220 const RunTimeOperandInfo* bw_output_gate_bias_;
222 const RunTimeOperandInfo* bw_projection_weights_;
223 const RunTimeOperandInfo* bw_projection_bias_;
225 const RunTimeOperandInfo* bw_input_layer_norm_weights_;
226 const RunTimeOperandInfo* bw_forget_layer_norm_weights_;
227 const RunTimeOperandInfo* bw_cell_layer_norm_weights_;
228 const RunTimeOperandInfo* bw_output_layer_norm_weights_;
230 const RunTimeOperandInfo* bw_activation_state_;
231 const RunTimeOperandInfo* bw_cell_state_;
232 RunTimeOperandInfo* bw_output_;
234 RunTimeOperandInfo* fw_output_activation_state_;
235 RunTimeOperandInfo* fw_output_cell_state_;
236 RunTimeOperandInfo* bw_output_activation_state_;
237 RunTimeOperandInfo* bw_output_cell_state_;