Searched defs:reshape3 (Results 1 – 3 of 3) sorted by relevance
511 auto reshape3 = in TEST_F() local
349 HloInstruction* reshape3 = builder.AddInstruction( in TEST_F() local
208 auto reshape3 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local