From 9291358d29a12ed69aef1cd238dbae673f150692 Mon Sep 17 00:00:00 2001 From: zyfncg Date: Mon, 18 Apr 2022 20:24:24 +0800 Subject: [PATCH] add data transform config for shape and size (#41909) --- python/paddle/utils/code_gen/api.yaml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/paddle/utils/code_gen/api.yaml b/python/paddle/utils/code_gen/api.yaml index 41b5fc26fa..351eca5a6b 100644 --- a/python/paddle/utils/code_gen/api.yaml +++ b/python/paddle/utils/code_gen/api.yaml @@ -1813,6 +1813,8 @@ func : ShapeInferMeta kernel : func : shape, shape_sr + data_transform: + skip_transform : input # shard_index - api : shard_index @@ -1889,6 +1891,8 @@ func : SizeInferMeta kernel : func : size + data_transform: + skip_transform : x - api : slice args : (Tensor input, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis) -- GitLab