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