Searched refs:cell_scratch (Results 1 – 1 of 1) sorted by relevance
797 float* cell_scratch = nullptr; in LSTMStep() local801 cell_scratch = scratch_buffer_buffer; in LSTMStep()802 forget_gate_scratch = cell_scratch + n_cell * n_batch; in LSTMStep()803 output_gate_scratch = cell_scratch + 2 * n_cell * n_batch; in LSTMStep()806 cell_scratch = input_gate_scratch + n_cell * n_batch; in LSTMStep()829 std::fill_n(cell_scratch, n_cell * n_batch, 0.0f); in LSTMStep()861 cell_scratch); in LSTMStep()878 cell_scratch); in LSTMStep()923 tflite::tensor_utils::MeanStddevNormalization(cell_scratch, cell_scratch, n_cell, n_batch); in LSTMStep()925 cell_scratch, n_batch, cell_scratch); in LSTMStep()[all …]