Searched defs:transpose1 (Results 1 – 3 of 3) sorted by relevance
580 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local
1349 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm1); in TEST_F() local1451 Output transpose1 = ops::Transpose(s, inputs, ops::Const(s, {1, 0})); in TEST_F() local1489 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm); in TEST_F() local1514 Output transpose1 = ops::Transpose( in TEST_F() local
2768 HloInstruction* transpose1 = in TEST_F() local