Searched defs:AllToAll (Results 1 – 5 of 5) sorted by relevance
42 AllToAll::AllToAll() in AllToAll() function in tensorflow::AllToAll
115 ## AllToAll section in Operation Semantics
434 TEST_F(XlaBuilderTest, AllToAll) { in TEST_F() argument
2946 XlaOp XlaBuilder::AllToAll(XlaOp operand, int64_t split_dimension, in AllToAll() function in xla::XlaBuilder4539 XlaOp AllToAll(const XlaOp operand, int64_t split_dimension, in AllToAll() function
17652 func AllToAll(scope *Scope, input tf.Output, group_assignment tf.Output, concat_dimension int64, sp… func