Searched defs:output_buffer (Results 1 – 1 of 1) sorted by relevance
425 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer, in LSTMEvalFloat32()521 bool LSTMCell::LSTMEvalFloat16( in LSTMEvalFloat16()785 float* cell_state_out_buffer, float* output_buffer, float* scratch_buffer_buffer) { in LSTMStep()