Browse Source

modify apply_momentum output var

tags/v0.3.0-alpha
changzherui 5 years ago
parent
commit
e01692adda
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/_op_impl/tbe/apply_momentum.py

+ 1
- 1
mindspore/ops/_op_impl/tbe/apply_momentum.py 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,


Loading…
Cancel
Save