提交 09f7d539 编写于 作者: S Smit Hinsu 提交者: TensorFlower Gardener

Mark input and output shapes attribute as deprecated in TRTEngineOp

PiperOrigin-RevId: 258412619
上级 da3a241a
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册