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.cpp424 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer, in LSTMEvalFloat32() argument
499 output_state_out_buffer, cell_state_out_buffer, outputCurrentTimeStep, in LSTMEvalFloat32()
506 outputStateInCurrentTimeStep.assign(output_state_out_buffer, in LSTMEvalFloat32()
507 output_state_out_buffer + batchSize * outputSize); in LSTMEvalFloat32()
544 const _Float16* output_layer_norm_weights_buffer, _Float16* output_state_out_buffer, in LSTMEvalFloat16() argument
670 convertFloat16ToFloat32(output_state_out_buffer, &output_state_out_float32); in LSTMEvalFloat16()
757 convertFloat32ToFloat16(output_state_out_float32, output_state_out_buffer); in LSTMEvalFloat16()
784 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer, in LSTMStep() argument
990 std::copy_n(output_buffer, n_batch * n_output, output_state_out_buffer); in LSTMStep()