Searched defs:transpose1 (Results 1 – 4 of 4) sorted by relevance
149 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() local
2223 HloInstruction* transpose1 = in TEST_F() local
1366 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm1); in TEST_F() local1436 Output transpose1 = ops::Transpose(s, inputs, ops::Const(s, {1, 0})); in TEST_F() local1474 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm); in TEST_F() local1499 Output transpose1 = ops::Transpose( in TEST_F() local
581 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local