Home
last modified time | relevance | path

Searched refs:output_state_out_buffer (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTM.h128 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer,
156 const _Float16* output_layer_norm_weights_buffer, _Float16* output_state_out_buffer,
181 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer,
DLSTM.cpp381 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer, in LSTMEvalFloat32() argument
456 output_state_out_buffer, cell_state_out_buffer, outputCurrentTimeStep, in LSTMEvalFloat32()
463 outputStateInCurrentTimeStep.assign(output_state_out_buffer, in LSTMEvalFloat32()
464 output_state_out_buffer + batchSize * outputSize); in LSTMEvalFloat32()
501 const _Float16* output_layer_norm_weights_buffer, _Float16* output_state_out_buffer, in LSTMEvalFloat16() argument
627 convertFloat16ToFloat32(output_state_out_buffer, &output_state_out_float32); in LSTMEvalFloat16()
714 convertFloat32ToFloat16(output_state_out_float32, output_state_out_buffer); in LSTMEvalFloat16()
741 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer, in LSTMStep() argument
948 tflite::tensor_utils::CopyVector(output_buffer, n_batch * n_output, output_state_out_buffer); in LSTMStep()