Searched refs:original_reshape (Results 1 – 1 of 1) sorted by relevance
3516 HloInstruction* original_reshape = in TEST_F() local3521 ShapeUtil::MakeShape(F32, {dim0 - 2, dim1, dim2}), original_reshape, in TEST_F()3542 HloInstruction* original_reshape = in TEST_F() local3547 ShapeUtil::MakeShape(F32, {960, 512}), original_reshape, in TEST_F()