Lines Matching refs:RunTimeOperandInfo

28 inline T *GetBuffer(RunTimeOperandInfo* operand) {  in GetBuffer()
33 inline const T *GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
40 std::vector<RunTimeOperandInfo>& operands) { in LSTMCell()
82 const Operation &operation, std::vector<RunTimeOperandInfo> &operands, in CheckInputTensorDimensions()
97 const RunTimeOperandInfo *input_to_input_weights = in CheckInputTensorDimensions()
105 const RunTimeOperandInfo *input_to_forget_weights = in CheckInputTensorDimensions()
111 const RunTimeOperandInfo *input_to_cell_weights = in CheckInputTensorDimensions()
117 const RunTimeOperandInfo *recurrent_to_input_weights = in CheckInputTensorDimensions()
125 const RunTimeOperandInfo *recurrent_to_forget_weights = in CheckInputTensorDimensions()
131 const RunTimeOperandInfo *recurrent_to_cell_weights = in CheckInputTensorDimensions()
146 const RunTimeOperandInfo *cell_to_input_weights = in CheckInputTensorDimensions()
153 const RunTimeOperandInfo *cell_to_forget_weights = in CheckInputTensorDimensions()
160 const RunTimeOperandInfo *cell_to_output_weights = in CheckInputTensorDimensions()
179 const RunTimeOperandInfo* input_gate_bias = in CheckInputTensorDimensions()
188 const RunTimeOperandInfo *forget_gate_bias = in CheckInputTensorDimensions()
193 const RunTimeOperandInfo *cell_bias = in CheckInputTensorDimensions()
198 const RunTimeOperandInfo *output_gate_bias = in CheckInputTensorDimensions()
203 const RunTimeOperandInfo *projection_weights = in CheckInputTensorDimensions()
211 const RunTimeOperandInfo *projection_bias = in CheckInputTensorDimensions()
231 std::vector<RunTimeOperandInfo> &operands, in Prepare()
243 const RunTimeOperandInfo *input = in Prepare()
249 const RunTimeOperandInfo *input_to_output_weights = in Prepare()
255 const RunTimeOperandInfo *recurrent_to_output_weights = in Prepare()
285 const RunTimeOperandInfo *input_to_input_weights = in Prepare()