Searched refs:cell_state_out_buffer (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 425 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer, in LSTMEvalFloat32() argument
499 output_state_out_buffer, cell_state_out_buffer, outputCurrentTimeStep, in LSTMEvalFloat32()
508 cellStateInCurrentTimeStep.assign(cell_state_out_buffer, in LSTMEvalFloat32()
509 cell_state_out_buffer + batchSize * numCells); in LSTMEvalFloat32()
545 _Float16* cell_state_out_buffer, _Float16* output_buffer, _Float16* scratch_buffer_buffer, in LSTMEvalFloat16()
672 convertFloat16ToFloat32(cell_state_out_buffer, &cell_state_out_float32); in LSTMEvalFloat16()
758 convertFloat32ToFloat16(cell_state_out_float32, cell_state_out_buffer); in LSTMEvalFloat16()
785 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer) { in LSTMStep() argument
932 n_batch * n_cell, cell_state_out_buffer); in LSTMStep()
939 cell_scratch, forget_gate_scratch, n_batch * n_cell, cell_state_out_buffer); in LSTMStep()
[all …]
|
| D | LSTM.h | 129 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer,
157 _Float16* cell_state_out_buffer, _Float16* output_buffer,
182 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer);
|