Function at::_scaled_mm_v2_outf

• Defined in File Functions.h

Function Documentation

inline at::Tensor &at::_scaled_mm_v2_outf(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> &bias, ::std::optional<at::ScalarType> out_dtype, at::IntArrayRef contraction_dim, bool use_fast_accum, at::Tensor &out)