Searched defs:ReduceScatterOptions (Results 1 – 3 of 3) sorted by relevance
152 struct ReduceScatterOptions { struct153 ReduceOp reduceOp = ReduceOp::SUM;154 std::chrono::milliseconds timeout = kUnsetTimeout;155 bool asyncOp = true;
123 const ReduceScatterOptions& /* opts */ = in reduce_scatter()131 const ReduceScatterOptions& /* opts */ = in _reduce_scatter_base()139 const ReduceScatterOptions& /* opts */ = in reduce_scatter_tensor_coalesced()
204 const ReduceScatterOptions& /* opts */ = ReduceScatterOptions()) { in reduce_scatter()214 const ReduceScatterOptions& /* opts */ = ReduceScatterOptions()) { in _reduce_scatter_base()229 const ReduceScatterOptions& /* opts */ = ReduceScatterOptions()) { in reduce_scatter_tensor_coalesced()