From e01692addaa2753dbf6b8565995b429c04463e8f Mon Sep 17 00:00:00 2001 From: changzherui Date: Tue, 5 May 2020 17:08:26 +0800 Subject: [PATCH] modify apply_momentum output var --- mindspore/ops/_op_impl/tbe/apply_momentum.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/_op_impl/tbe/apply_momentum.py b/mindspore/ops/_op_impl/tbe/apply_momentum.py index 42ce9d0e41..920a48e48d 100644 --- a/mindspore/ops/_op_impl/tbe/apply_momentum.py +++ b/mindspore/ops/_op_impl/tbe/apply_momentum.py @@ -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,