未验证 提交 9291358d 编写于 作者: Z zyfncg 提交者: GitHub

add data transform config for shape and size (#41909)

上级 bc1c3e3e
...@@ -1813,6 +1813,8 @@ ...@@ -1813,6 +1813,8 @@
func : ShapeInferMeta func : ShapeInferMeta
kernel : kernel :
func : shape, shape_sr func : shape, shape_sr
data_transform:
skip_transform : input
# shard_index # shard_index
- api : shard_index - api : shard_index
...@@ -1889,6 +1891,8 @@ ...@@ -1889,6 +1891,8 @@
func : SizeInferMeta func : SizeInferMeta
kernel : kernel :
func : size func : size
data_transform:
skip_transform : x
- api : slice - api : slice
args : (Tensor input, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis) args : (Tensor input, int64_t[] axes, IntArray starts, IntArray ends, int64_t[] infer_flags, int64_t[] decrease_axis)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册