Searched defs:forget_layer_norm_weights (Results 1 – 1 of 1) sorted by relevance
128 …TMParams &set_layer_normalization_params(T *input_layer_norm_weights, T *forget_layer_norm_weights, in set_layer_normalization_params()240 T *forget_layer_norm_weights() const in forget_layer_norm_weights() function