Searched refs:output_state_out_buffer (Results 1 – 2 of 2) sorted by relevance
/packages/modules/NeuralNetworks/common/types/operations/include/ |
D | LSTM.h | 127 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/ |
D | LSTM.cpp | 424 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()
|