Skip to content

M
mindspore

正统之独孤求败 / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 0
Fork 0
M
mindspore

体验新版 GitCode，发现更多精彩内容 >>

提交 e01692ad 编写于 作者: C changzherui
浏览文件
操作

modify apply_momentum output var

上级 84c029d1
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
mindspore/ops/_op_impl/tbe/apply_momentum.py
浏览文件 @ e01692ad
......@@ -29,7 +29,7 @@ apply_momentum_op_info = TBERegOp("ApplyMomentum") \
.input(2, "lr", False, "required", "all") \
.input(3, "grad", False, "required", "all") \
.input(4, "momentum", False, "required", "all") \
.output(0, "var", False, "required", "all") \
.output(0, "out", False, "required", "all") \
.dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
DataType.F16_Default, DataType.F16_Default) \
.dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_5HD,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或