26 lines
1.1 KiB
Diff
26 lines
1.1 KiB
Diff
From d8e07ff51f9e709399b8c553290836fb308e45ed Mon Sep 17 00:00:00 2001
|
|
From: geetachavan1 <53313357+geetachavan1@users.noreply.github.com>
|
|
Date: Tue, 27 Jul 2021 16:08:12 -0700
|
|
Subject: [PATCH 1/1] Update shape_refiner.cc
|
|
|
|
---
|
|
tensorflow/core/common_runtime/shape_refiner.cc | 2 +-
|
|
1 file changed, 1 insertion(+), 1 deletion(-)
|
|
|
|
diff --git a/tensorflow/core/common_runtime/shape_refiner.cc b/tensorflow/core/common_runtime/shape_refiner.cc
|
|
index 6a7d1eadfb6..906bd14f96c 100644
|
|
--- a/tensorflow/core/common_runtime/shape_refiner.cc
|
|
+++ b/tensorflow/core/common_runtime/shape_refiner.cc
|
|
@@ -132,7 +132,7 @@ Status InferShapesForFunctionSubNode(const Node* node, ShapeRefiner* refiner,
|
|
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(
|
|
--
|
|
2.27.0
|
|
|