Function at::_fused_rms_norm

• Defined in File Functions.h

Function Documentation

inline at::Tensor at::_fused_rms_norm(const at::Tensor &input, int64_t normalized_shape_ndim, const at::Tensor &weight, double eps)