Lines Matching refs:RunTimeOperandInfo

46 struct RunTimeOperandInfo;
51     LSTMCell(const Operation& operation, RunTimeOperandInfo* operands);
53     bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* scratchShape,
185             const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights,
186             const RunTimeOperandInfo* input_to_forget_weights,
187             const RunTimeOperandInfo* input_to_cell_weights,
188             const RunTimeOperandInfo* input_to_output_weights,
189             const RunTimeOperandInfo* recurrent_to_input_weights,
190             const RunTimeOperandInfo* recurrent_to_forget_weights,
191             const RunTimeOperandInfo* recurrent_to_cell_weights,
192             const RunTimeOperandInfo* recurrent_to_output_weights,
193             const RunTimeOperandInfo* cell_to_input_weights,
194             const RunTimeOperandInfo* cell_to_forget_weights,
195             const RunTimeOperandInfo* cell_to_output_weights,
196             const RunTimeOperandInfo* input_gate_bias, const RunTimeOperandInfo* forget_gate_bias,
197             const RunTimeOperandInfo* cell_bias, const RunTimeOperandInfo* output_gate_bias,
198             const RunTimeOperandInfo* projection_weights, const RunTimeOperandInfo* projection_bias,
199             const RunTimeOperandInfo* input_layer_norm_weights,
200             const RunTimeOperandInfo* forget_layer_norm_weights,
201             const RunTimeOperandInfo* cell_layer_norm_weights,
202             const RunTimeOperandInfo* output_layer_norm_weights, uint32_t n_input,
207     const RunTimeOperandInfo* input_;
209     const RunTimeOperandInfo* input_to_input_weights_;
210     const RunTimeOperandInfo* input_to_forget_weights_;
211     const RunTimeOperandInfo* input_to_cell_weights_;
212     const RunTimeOperandInfo* input_to_output_weights_;
214     const RunTimeOperandInfo* recurrent_to_input_weights_;
215     const RunTimeOperandInfo* recurrent_to_forget_weights_;
216     const RunTimeOperandInfo* recurrent_to_cell_weights_;
217     const RunTimeOperandInfo* recurrent_to_output_weights_;
219     const RunTimeOperandInfo* cell_to_input_weights_;
220     const RunTimeOperandInfo* cell_to_forget_weights_;
221     const RunTimeOperandInfo* cell_to_output_weights_;
223     const RunTimeOperandInfo* input_gate_bias_;
224     const RunTimeOperandInfo* forget_gate_bias_;
225     const RunTimeOperandInfo* cell_bias_;
226     const RunTimeOperandInfo* output_gate_bias_;
228     const RunTimeOperandInfo* projection_weights_;
229     const RunTimeOperandInfo* projection_bias_;
231     const RunTimeOperandInfo* output_state_in_;
232     const RunTimeOperandInfo* cell_state_in_;
234     const RunTimeOperandInfo* input_layer_norm_weights_;
235     const RunTimeOperandInfo* forget_layer_norm_weights_;
236     const RunTimeOperandInfo* cell_layer_norm_weights_;
237     const RunTimeOperandInfo* output_layer_norm_weights_;
239     RunTimeOperandInfo* output_state_out_;
240     RunTimeOperandInfo* cell_state_out_;
241     RunTimeOperandInfo* output_;
243     RunTimeOperandInfo* scratch_buffer_;