Lines Matching refs:output_buffer
425 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer, in LSTMEvalFloat32() argument
467 float* outputData = timeMajor ? output_buffer : transposedOutput.data(); in LSTMEvalFloat32()
514 output_buffer); in LSTMEvalFloat32()
545 _Float16* cell_state_out_buffer, _Float16* output_buffer, _Float16* scratch_buffer_buffer, in LSTMEvalFloat16()
675 convertFloat16ToFloat32(output_buffer, &output_float32); in LSTMEvalFloat16()
759 convertFloat32ToFloat16(output_float32, output_buffer); in LSTMEvalFloat16()
785 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer) { in LSTMStep() argument
975 output_buffer); in LSTMStep()
977 std::fill_n(output_buffer, n_batch * n_output, 0.0f); in LSTMStep()
981 output_buffer, in LSTMStep()
984 tflite::tensor_utils::ClipVector(output_buffer, n_batch * n_output, params.proj_clip, in LSTMStep()
985 output_buffer); in LSTMStep()
988 std::copy_n(output_gate_scratch, n_batch * n_output, output_buffer); in LSTMStep()
990 std::copy_n(output_buffer, n_batch * n_output, output_state_out_buffer); in LSTMStep()