Searched defs:NCCLPreMulSumSupplement (Results 1 – 1 of 1) sorted by relevance
23 struct NCCLPreMulSumSupplement : _SupplementBase { struct24 double double_factor{0.0};25 at::Tensor tensor_factor;26 NCCLPreMulSumSupplement(double f) : double_factor{f} {} in NCCLPreMulSumSupplement() function27 NCCLPreMulSumSupplement(at::Tensor t) : tensor_factor{std::move(t)} { in NCCLPreMulSumSupplement() function