Lines Matching refs:cell_scratch
797 float* cell_scratch = nullptr; in LSTMStep() local
801 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()
820 cell_scratch); in LSTMStep()
829 std::fill_n(cell_scratch, n_cell * n_batch, 0.0f); in LSTMStep()
844 cell_scratch, /*result_stride*/ 1); in LSTMStep()
863 cell_scratch, /*result_stride=*/1); in LSTMStep()
881 cell_scratch, /*result_stride*/ 1); in LSTMStep()
926 tflite::tensor_utils::MeanStddevNormalization(cell_scratch, cell_scratch, n_cell, n_batch); in LSTMStep()
928 cell_scratch, n_batch, cell_scratch); in LSTMStep()
929 tflite::tensor_utils::VectorBatchVectorAdd(cell_bias_buffer, n_cell, n_batch, cell_scratch); in LSTMStep()
933 tflite::tensor_utils::ApplyActivationToVector(cell_scratch, n_batch * n_cell, params.activation, in LSTMStep()
934 cell_scratch); in LSTMStep()
939 cell_scratch, forget_gate_scratch, n_batch * n_cell, cell_state_out_buffer); in LSTMStep()
942 cell_scratch, input_gate_scratch, n_batch * n_cell, cell_state_out_buffer); in LSTMStep()
967 params.activation, cell_scratch); in LSTMStep()
968 tflite::tensor_utils::VectorVectorCwiseProduct(output_gate_scratch, cell_scratch, in LSTMStep()