Searched refs:_check_nested_layer_norm_inputs (Results 1 – 3 of 3) sorted by relevance
28 C10_ALWAYS_INLINE std::pair<int64_t, int64_t> _check_nested_layer_norm_inputs( in _check_nested_layer_norm_inputs() function
211 auto M_N = _check_nested_layer_norm_inputs( in layer_norm_backward_nested()
161 auto M_N = _check_nested_layer_norm_inputs(*nt_input, normalized_shape, weight, bias); in nested_layer_norm()