Searched refs:input_gate_scratch (Results 1 – 1 of 1) sorted by relevance
796 float* input_gate_scratch = nullptr; in LSTMStep() local805 input_gate_scratch = scratch_buffer_buffer; in LSTMStep()806 cell_scratch = input_gate_scratch + n_cell * n_batch; in LSTMStep()807 forget_gate_scratch = input_gate_scratch + 2 * n_cell * n_batch; in LSTMStep()808 output_gate_scratch = input_gate_scratch + 3 * n_cell * n_batch; in LSTMStep()815 input_gate_scratch); in LSTMStep()826 std::fill_n(input_gate_scratch, n_cell * n_batch, 0.0f); in LSTMStep()837 input_gate_scratch, /*result_stride*/ 1); in LSTMStep()854 n_batch, input_gate_scratch, in LSTMStep()873 n_batch, input_gate_scratch, in LSTMStep()[all …]