From 9291358d29a12ed69aef1cd238dbae673f150692 Mon Sep 17 00:00:00 2001
From: zyfncg <zhangyunfei07@baidu.com>
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 41b5fc26fa9..351eca5a6b8 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