Merge pull request #50775 from tensorflow/mm-cherrypick-fix-onr2.4

Fix build after previous cherrypick (#50773).
This commit is contained in:
Mihai Maruseac 2021-07-14 13:47:47 -07:00 committed by GitHub
commit ecee625c3d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -135,7 +135,7 @@ Status ShapeRefiner::InferShapesForFunctionSubNode(
TF_RETURN_IF_ERROR(
outer_context->MakeShapeFromShapeProto(proto, &handle));
copied_shapes_and_types.push_back(
ShapeAndType(handle, shape_and_type.dtype, shape_and_type.type));
ShapeAndType(handle, shape_and_type.dtype, shape_and_type.specialized_type));
}
outer_context->set_output_handle_shapes_and_types(