Searched refs:couple_input_forget_gates_ (Results 1 – 2 of 2) sorted by relevance
194 input2input_ = couple_input_forget_gates_ in SetWeightForInputToInputGate()204 int input_forget_start = couple_input_forget_gates_ ? n_cell_ : 2 * n_cell_; in SetWeightForInputToForgetGate()214 couple_input_forget_gates_ ? 2 * n_cell_ : 3 * n_cell_; in SetWeightForInputToOutputGate()232 rec2input_ = couple_input_forget_gates_ in SetWeightForRecurrentToInputGate()242 int rec_forget_start = couple_input_forget_gates_ ? n_cell_ : 2 * n_cell_; in SetWeightForRecurrentToForgetGate()251 int rec_output_start = couple_input_forget_gates_ ? 2 * n_cell_ : 3 * n_cell_; in SetWeightForRecurrentToOutputGate()269 couple_input_forget_gates_ in SetBiasToInputGate()277 int bias_forget_start = couple_input_forget_gates_ ? n_cell_ : 2 * n_cell_; in SetBiasToForgetGate()286 couple_input_forget_gates_ ? 2 * n_cell_ : 3 * n_cell_; in SetBiasToOutputGate()467 couple_input_forget_gates_ = in InitializeFromFuncAttributes()[all …]
52 couple_input_forget_gates_(false), in ConvertLSTMCellSimpleToFusedLSTM()109 bool couple_input_forget_gates_; variable