Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/operations/include/
DLSTM.h127 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer,
155 const _Float16* output_layer_norm_weights_buffer, _Float16* output_state_out_buffer,
180 const float* output_layer_norm_weights_buffer, float* output_state_out_buffer,
/packages/modules/NeuralNetworks/common/cpu_operations/
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
988 std::copy_n(output_buffer, n_batch * n_output, output_state_out_buffer); in LSTMStep()