m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modify apply_momentum output var

Browse Source
pull/981/head
changzherui 5 years ago
parent 84c029d1ca
commit e01692adda
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
mindspore/ops/_op_impl/tbe/apply_momentum.py
Unescape View File

@ -29,7 +29,7 @@ apply_momentum_op_info = TBERegOp("ApplyMomentum") \
.input(2, "lr", False, "required", "all") \ .input(2, "lr", False, "required", "all") \
.input(3, "grad", False, "required", "all") \ .input(3, "grad", False, "required", "all") \
.input(4, "momentum", 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, .dtype_format(DataType.F16_Default, DataType.F16_Default, 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, .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_5HD,

Write Preview
Loading…
Cancel
Save