Searched defs:CollectivePermute (Results 1 – 6 of 6) sorted by relevance
603 ## CollectivePermute section in Operation Semantics
418 TEST_F(HloControlFlowFlatteningTest, CollectivePermute) { in TEST_F() argument
1988 struct CollectivePermute { struct1998 static CollectivePermute Handler() { return CollectivePermute(); } in Handler()2065 static bool CollectivePermute(runtime::KernelContext* ctx, void** args, in CollectivePermute() function
489 TEST_F(XlaBuilderTest, CollectivePermute) { in TEST_F() argument
3226 XlaOp XlaBuilder::CollectivePermute( in CollectivePermute() function in xla::XlaBuilder4714 XlaOp CollectivePermute( in CollectivePermute() function
6020 func CollectivePermute(scope *Scope, input tf.Output, source_target_pairs tf.Output) (output tf.Out… func