Lines Matching refs:RunTimeOperandInfo

38 inline T* GetBuffer(RunTimeOperandInfo* operand) {  in GetBuffer()
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
48 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer()
54 LSTMCell::LSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in LSTMCell()
115 static RunTimeOperandInfo no_value; in LSTMCell()
133 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights, in CheckInputTensorDimensions()
134 const RunTimeOperandInfo* input_to_forget_weights, in CheckInputTensorDimensions()
135 const RunTimeOperandInfo* input_to_cell_weights, in CheckInputTensorDimensions()
136 const RunTimeOperandInfo* input_to_output_weights, in CheckInputTensorDimensions()
137 const RunTimeOperandInfo* recurrent_to_input_weights, in CheckInputTensorDimensions()
138 const RunTimeOperandInfo* recurrent_to_forget_weights, in CheckInputTensorDimensions()
139 const RunTimeOperandInfo* recurrent_to_cell_weights, in CheckInputTensorDimensions()
140 const RunTimeOperandInfo* recurrent_to_output_weights, in CheckInputTensorDimensions()
141 const RunTimeOperandInfo* cell_to_input_weights, in CheckInputTensorDimensions()
142 const RunTimeOperandInfo* cell_to_forget_weights, in CheckInputTensorDimensions()
143 const RunTimeOperandInfo* cell_to_output_weights, const RunTimeOperandInfo* input_gate_bias, in CheckInputTensorDimensions()
144 const RunTimeOperandInfo* forget_gate_bias, const RunTimeOperandInfo* cell_bias, in CheckInputTensorDimensions()
145 const RunTimeOperandInfo* output_gate_bias, const RunTimeOperandInfo* projection_weights, in CheckInputTensorDimensions()
146 const RunTimeOperandInfo* projection_bias, in CheckInputTensorDimensions()
147 const RunTimeOperandInfo* input_layer_norm_weights, in CheckInputTensorDimensions()
148 const RunTimeOperandInfo* forget_layer_norm_weights, in CheckInputTensorDimensions()
149 const RunTimeOperandInfo* cell_layer_norm_weights, in CheckInputTensorDimensions()
150 const RunTimeOperandInfo* output_layer_norm_weights, uint32_t n_input, uint32_t n_output, in CheckInputTensorDimensions()
303 bool LSTMCell::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()