Searched defs:ReduceScatter (Results 1 – 6 of 6) sorted by relevance
2146 ## ReduceScatter section in Operation Semantics
1790 struct ReduceScatter { struct1798 static ReduceScatter Handler() { return ReduceScatter(); } in Handler()1836 static bool ReduceScatter(runtime::KernelContext* ctx, void** args, in ReduceScatter() function
191 class ReduceScatter: class
1021 XLA_TEST_F(CollectiveOpsTest, ReduceScatter) { in XLA_TEST_F() argument
410 TEST_F(XlaBuilderTest, ReduceScatter) { in TEST_F() argument
3017 XlaOp XlaBuilder::ReduceScatter( in ReduceScatter() function in xla::XlaBuilder4679 XlaOp ReduceScatter(const XlaOp operand, const XlaComputation& computation, in ReduceScatter() function