Searched defs:transpose1 (Results 1 – 4 of 4) sorted by relevance
174 const HloInstruction* transpose1 = nullptr; in TEST_P() local
3176 HloInstruction* transpose1 = in TEST_F() local
578 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local
1808 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm1); in TEST_F() local1910 Output transpose1 = ops::Transpose(s, inputs, ops::Const(s, {1, 0})); in TEST_F() local1948 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm); in TEST_F() local1973 Output transpose1 = ops::Transpose( in TEST_F() local