Function at::_scaled_grouped_mm_v2

• Defined in File Functions.h

Function Documentation

inline at::Tensor at::_scaled_grouped_mm_v2(const at::Tensor &self, const at::Tensor &mat2, at::TensorList scale_a, at::IntArrayRef recipe_a, at::IntArrayRef swizzle_a, at::TensorList scale_b, at::IntArrayRef recipe_b, at::IntArrayRef swizzle_b, const ::std::optional<at::Tensor> &offs = {}, const ::std::optional<at::Tensor> &bias = {}, ::std::optional<at::ScalarType> out_dtype = ::std::nullopt, at::IntArrayRef contraction_dim = {}, bool use_fast_accum = false)