Searched refs:shape3 (Results 1 – 1 of 1) sorted by relevance
495 auto shape3 = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local510 builder.AddInstruction(HloInstruction::CreateReshape(shape3, param2)); in TEST_F()512 builder.AddInstruction(HloInstruction::CreateReshape(shape3, add0)); in TEST_F()513 builder.AddInstruction(HloInstruction::CreateBinary(shape3, HloOpcode::kAdd, in TEST_F()