• Home
  • Raw
  • Download

Lines Matching refs:RunTimeOperandInfo

43 inline T* GetBuffer(RunTimeOperandInfo* operand) {  in GetBuffer()
48 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
53 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer()
59 LSTMCell::LSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in LSTMCell()
120 static RunTimeOperandInfo no_value; in LSTMCell()
138 const RunTimeOperandInfo* /*input_*/, const RunTimeOperandInfo* input_to_input_weights, in CheckInputTensorDimensions() argument
139 const RunTimeOperandInfo* input_to_forget_weights, in CheckInputTensorDimensions()
140 const RunTimeOperandInfo* input_to_cell_weights, in CheckInputTensorDimensions()
141 const RunTimeOperandInfo* /*input_to_output_weights*/, in CheckInputTensorDimensions()
142 const RunTimeOperandInfo* recurrent_to_input_weights, in CheckInputTensorDimensions()
143 const RunTimeOperandInfo* recurrent_to_forget_weights, in CheckInputTensorDimensions()
144 const RunTimeOperandInfo* recurrent_to_cell_weights, in CheckInputTensorDimensions()
145 const RunTimeOperandInfo* /*recurrent_to_output_weights*/, in CheckInputTensorDimensions()
146 const RunTimeOperandInfo* cell_to_input_weights, in CheckInputTensorDimensions()
147 const RunTimeOperandInfo* cell_to_forget_weights, in CheckInputTensorDimensions()
148 const RunTimeOperandInfo* cell_to_output_weights, const RunTimeOperandInfo* input_gate_bias, in CheckInputTensorDimensions()
149 const RunTimeOperandInfo* forget_gate_bias, const RunTimeOperandInfo* cell_bias, in CheckInputTensorDimensions()
150 const RunTimeOperandInfo* output_gate_bias, const RunTimeOperandInfo* projection_weights, in CheckInputTensorDimensions()
151 const RunTimeOperandInfo* projection_bias, in CheckInputTensorDimensions()
152 const RunTimeOperandInfo* input_layer_norm_weights, in CheckInputTensorDimensions()
153 const RunTimeOperandInfo* forget_layer_norm_weights, in CheckInputTensorDimensions()
154 const RunTimeOperandInfo* cell_layer_norm_weights, in CheckInputTensorDimensions()
155 const RunTimeOperandInfo* output_layer_norm_weights, uint32_t n_input, uint32_t n_output, in CheckInputTensorDimensions()
308 bool LSTMCell::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()