Searched defs:reshape1 (Results 1 – 6 of 6) sorted by relevance
49 auto reshape1 = in TEST_F() local118 auto reshape1 = in TEST_F() local145 auto reshape1 = in TEST_F() local193 auto reshape1 = variable276 auto reshape1 = variable368 auto reshape1 = variable397 auto reshape1 = in TEST_F() local430 auto reshape1 = in TEST_F() local505 auto reshape1 = in TEST_F() local
901 auto reshape1 = in TEST_F() local
1868 auto reshape1 = builder.AddInstruction( in TEST_F() local2712 HloInstruction* reshape1 = in TEST_F() local2905 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
350 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local365 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local380 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local395 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local410 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local425 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local440 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local455 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local562 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
192 auto reshape1 = Reshape(slice1, {}); in TEST_F() local
432 HloInstruction* reshape1 = builder.AddInstruction( in TEST_F() local