Searched refs:forget_gate_scratch (Results 1 – 1 of 1) sorted by relevance
798 float* forget_gate_scratch = nullptr; in LSTMStep() local802 forget_gate_scratch = cell_scratch + n_cell * n_batch; in LSTMStep()807 forget_gate_scratch = input_gate_scratch + 2 * n_cell * n_batch; in LSTMStep()818 forget_gate_scratch); in LSTMStep()828 std::fill_n(forget_gate_scratch, n_cell * n_batch, 0.0f); in LSTMStep()841 forget_gate_scratch, /*result_stride*/ 1); in LSTMStep()860 forget_gate_scratch, /*result_stride=*/1); in LSTMStep()878 forget_gate_scratch, /*result_stride*/ 1); in LSTMStep()910 n_batch, forget_gate_scratch); in LSTMStep()913 tflite::tensor_utils::MeanStddevNormalization(forget_gate_scratch, forget_gate_scratch, in LSTMStep()[all …]