Searched defs:CollectivePermute (Results 1 – 4 of 4) sorted by relevance
569 ## CollectivePermute section in Operation Semantics
438 TEST_F(XlaBuilderTest, CollectivePermute) { in TEST_F() argument
2996 XlaOp XlaBuilder::CollectivePermute( in CollectivePermute() function in xla::XlaBuilder4694 XlaOp CollectivePermute( in CollectivePermute() function
39408 func CollectivePermute(scope *Scope, input tf.Output, source_target_pairs tf.Output) (output tf.Out… func