From 09f7d539b95ad8e1e521811bc0395ce36bebd8b8 Mon Sep 17 00:00:00 2001 From: Smit Hinsu Date: Tue, 16 Jul 2019 11:59:27 -0700 Subject: [PATCH] Mark input and output shapes attribute as deprecated in TRTEngineOp PiperOrigin-RevId: 258412619 --- tensorflow/compiler/tf2tensorrt/ops/trt_engine_op.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/tf2tensorrt/ops/trt_engine_op.cc b/tensorflow/compiler/tf2tensorrt/ops/trt_engine_op.cc index a849aa418c6..b8f9058d8f6 100644 --- a/tensorflow/compiler/tf2tensorrt/ops/trt_engine_op.cc +++ b/tensorflow/compiler/tf2tensorrt/ops/trt_engine_op.cc @@ -33,8 +33,6 @@ namespace tensorflow { // key to cache the instantiated functions for different executor subgraphs. REGISTER_OP("TRTEngineOp") .Attr("serialized_segment: string") - .Attr("input_shapes: list(shape)") - .Attr("output_shapes: list(shape)") .Attr("segment_funcdef_name: string") .Attr("InT: list({int8,float16,float32,int32})") .Attr("OutT: list({int8,float16,float32,int32})") @@ -55,6 +53,8 @@ REGISTER_OP("TRTEngineOp") // Deprecated attributes. .Attr("cached_engine_batches: list(int) >= 0 = []") .Attr("fixed_input_size: bool = true") + .Attr("input_shapes: list(shape)") + .Attr("output_shapes: list(shape)") .Attr("static_engine: bool = true"); } // namespace tensorflow -- GitLab