Searched defs:AllToAll (Results 1 – 4 of 4) sorted by relevance
115 ## AllToAll section in Operation Semantics
408 TEST_F(XlaBuilderTest, AllToAll) { in TEST_F() argument
2857 XlaOp XlaBuilder::AllToAll(XlaOp operand, int64 split_dimension, in AllToAll() function in xla::XlaBuilder4677 XlaOp AllToAll(const XlaOp operand, int64 split_dimension, in AllToAll() function
51322 func AllToAll(scope *Scope, input tf.Output, group_assignment tf.Output, concat_dimension int64, sp… func