Function at::_sparse_mm_reduce_impl_backward#

Function Documentation#

inline ::std::tuple<at::Tensor, at::Tensor> at::_sparse_mm_reduce_impl_backward(const at::Tensor &self, const at::Tensor &grad_out, const at::Tensor &weight, c10::string_view reduce, const at::Tensor &arg_out, ::std::array<bool, 2> output_mask)#