Searched defs:AllreduceOptions (Results 1 – 3 of 3) sorted by relevance
121 struct AllreduceOptions { struct122 ReduceOp reduceOp = ReduceOp::SUM;123 std::chrono::milliseconds timeout = kUnsetTimeout;124 std::optional<at::Tensor> sparseIndices = std::nullopt;
32 const AllreduceOptions& /* opts */ = AllreduceOptions()) override { in allreduce()38 const AllreduceOptions& /* opts */ = AllreduceOptions()) override { in allreduce_sparse()
92 const AllreduceOptions& /* opts */ = AllreduceOptions()) { in allreduce()100 const AllreduceOptions& /* opts */ = AllreduceOptions()) { in allreduce_sparse()