Home
last modified time | relevance | path

Searched refs:input_forget_start (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc204 int input_forget_start = couple_input_forget_gates_ ? n_cell_ : 2 * n_cell_; in SetWeightForInputToForgetGate() local
205 SmallVector<int64_t, 2> begin_i2f_values = {input_forget_start, 0}; in SetWeightForInputToForgetGate()