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