Searched refs:begin_forget_layer_norm_values (Results 1 – 1 of 1) sorted by relevance
562 SmallVector<int64_t, 1> begin_forget_layer_norm_values = {2 * n_cell_}; in SetForgetLayerNormCoefficients() local565 begin_forget_layer_norm_values, layer_norm_slice_shape_, in SetForgetLayerNormCoefficients()