Searched defs:ReduceScatter (Results 1 – 4 of 4) sorted by relevance
2090 ## ReduceScatter section in Operation Semantics
1096 XLA_TEST_F(CollectiveOpsTest, ReduceScatter) { in XLA_TEST_F() argument
408 TEST_F(XlaBuilderTest, ReduceScatter) { in TEST_F() argument
2886 XlaOp XlaBuilder::ReduceScatter( in ReduceScatter() function in xla::XlaBuilder4528 XlaOp ReduceScatter(const XlaOp operand, const XlaComputation& computation, in ReduceScatter() function