Searched refs:begin_rec2i_values (Results 1 – 1 of 1) sorted by relevance
231 SmallVector<int64_t, 2> begin_rec2i_values = {n_cell_, n_input_}; in SetWeightForRecurrentToInputGate() local235 weight_slice_shape_, begin_rec2i_values, in SetWeightForRecurrentToInputGate()