未验证 提交 df21f815 编写于 作者: H hong 提交者: GitHub

[NewIR] fix new ir affine grid bug (#55244)

* fix affine grid bug

* revert cummax
上级 a8cd12d2
...@@ -117,7 +117,7 @@ ...@@ -117,7 +117,7 @@
inputs : inputs :
input : Theta input : Theta
outputs : outputs :
out : Output output : Output
int_array: int_array:
output_shape : output_shape :
data_type : int data_type : int
......
...@@ -100,7 +100,7 @@ ...@@ -100,7 +100,7 @@
- op : affine_grid - op : affine_grid
args : (Tensor input, IntArray output_shape={}, bool align_corners=true) args : (Tensor input, IntArray output_shape={}, bool align_corners=true)
output : Tensor output : Tensor(output)
infer_meta : infer_meta :
func : AffineGridInferMeta func : AffineGridInferMeta
param : [input, output_shape, align_corners] param : [input, output_shape, align_corners]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册