Searched refs:original_reshape (Results 1 – 1 of 1) sorted by relevance
2756 HloInstruction* original_reshape = in TEST_F() local2761 ShapeUtil::MakeShape(F32, {dim0 - 2, dim1, dim2}), original_reshape, in TEST_F()2782 HloInstruction* original_reshape = in TEST_F() local2787 ShapeUtil::MakeShape(F32, {960, 512}), original_reshape, in TEST_F()