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

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

* fix affine grid bug

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