Cross Reference: /frameworks/ml/nn/common/operations/LSTM.cpp

Lines Matching defs:GetBuffer
28 inline T *GetBuffer(RunTimeOperandInfo* operand) {
33 inline const T *GetBuffer(const RunTimeOperandInfo* operand) {
332     tflite::tensor_utils::VectorBatchVectorAssign(GetBuffer<float>(input_gate_bias_),
335   tflite::tensor_utils::VectorBatchVectorAssign(GetBuffer<float>(forget_gate_bias_),
337   tflite::tensor_utils::VectorBatchVectorAssign(GetBuffer<float>(cell_bias_),
339   tflite::tensor_utils::VectorBatchVectorAssign(GetBuffer<float>(output_gate_bias_),
345         GetBuffer<float>(input_to_input_weights_), n_cell, n_input,
346         GetBuffer<float>(input_), n_batch, input_gate_scratch, /*result_stride*/1);
349       GetBuffer<float>(input_to_forget_weights_), n_cell, n_input,
350       GetBuffer<float>(input_), n_batch, forget_gate_scratch, /*result_stride*/1);
352       GetBuffer<float>(input_to_cell_weights_), n_cell, n_input,
353       GetBuffer<float>(input_), n_batch, cell_scratch, /*result_stride*/1);
355       GetBuffer<float>(input_to_output_weights_), n_cell, n_input,
356       GetBuffer<float>(input_), n_batch, output_gate_scratch, /*result_stride*/1);
361         GetBuffer<float>(recurrent_to_input_weights_), n_cell, n_output,
362         GetBuffer<float>(output_state_in_), n_batch, input_gate_scratch, /*result_stride*/1);
365       GetBuffer<float>(recurrent_to_forget_weights_), n_cell, n_output,
366       GetBuffer<float>(output_state_in_), n_batch, forget_gate_scratch, /*result_stride*/1);
368       GetBuffer<float>(recurrent_to_cell_weights_), n_cell, n_output,
369       GetBuffer<float>(output_state_in_), n_batch, cell_scratch, /*result_stride*/1);
371       GetBuffer<float>(recurrent_to_output_weights_), n_cell, n_output,
372       GetBuffer<float>(output_state_in_), n_batch, output_gate_scratch, /*result_stride*/1);
378           GetBuffer<float>(cell_to_input_weights_), n_cell,
379           GetBuffer<float>(cell_state_in_), n_batch, input_gate_scratch);
389         GetBuffer<float>(cell_to_forget_weights_), n_cell,
390         GetBuffer<float>(cell_state_in_), n_batch, forget_gate_scratch);
398       forget_gate_scratch, GetBuffer<float>(cell_state_in_), n_batch * n_cell,
399       GetBuffer<float>(cell_state_out_));
408         GetBuffer<float>(cell_state_out_));
412         GetBuffer<float>(cell_state_out_));
415     tflite::tensor_utils::ClipVector(GetBuffer<float>(cell_state_out_), n_batch * n_cell,
416                                      params_.cell_clip_, GetBuffer<float>(cell_state_out_));
422         GetBuffer<float>(cell_to_output_weights_), n_cell,
423         GetBuffer<float>(cell_state_out_), n_batch, output_gate_scratch);
427   tflite::tensor_utils::ApplyActivationToVector(GetBuffer<float>(cell_state_out_),
441       tflite::tensor_utils::VectorBatchVectorAssign(GetBuffer<float>(projection_bias_), n_output,
442                                                     n_batch, GetBuffer<float>(output_));
444       tflite::tensor_utils::ZeroVector(GetBuffer<float>(output_), n_batch * n_output);
447         GetBuffer<float>(projection_weights_), n_output, n_cell,
448         output_gate_scratch, n_batch, GetBuffer<float>(output_),
451       tflite::tensor_utils::ClipVector(GetBuffer<float>(output_), n_batch * n_output,
452                                params_.proj_clip_, GetBuffer<float>(output_));
456                              GetBuffer<float>(output_));
458   tflite::tensor_utils::CopyVector(GetBuffer<float>(output_), n_batch * n_output,
459                            GetBuffer<float>(output_state_out_));
OpenGrok
