From d691f1c5d7df067e7ba09d8c1c083971a0401f6b Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Thu, 21 May 2020 18:38:39 +0800 Subject: [PATCH] fixed batch_norm_grad --- mindspore/ops/_op_impl/tbe/batchnorm_grad.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/mindspore/ops/_op_impl/tbe/batchnorm_grad.py b/mindspore/ops/_op_impl/tbe/batchnorm_grad.py index 6063c0e750..973f3709e5 100644 --- a/mindspore/ops/_op_impl/tbe/batchnorm_grad.py +++ b/mindspore/ops/_op_impl/tbe/batchnorm_grad.py @@ -31,23 +31,22 @@ batch_norm_grad_op_info = TBERegOp("BatchNormGrad") \ .input(2, "scale", False, "required", "all") \ .input(3, "reserve_space_1", False, "required", "all") \ .input(4, "reserve_space_2", False, "required", "all") \ - .input(5, "reserve_space_3", False, "required", "all") \ .output(0, "x_backprop", False, "required", "all") \ .output(1, "scale_backprop", False, "required", "all") \ .output(2, "offset_backprop", False, "required", "all") \ .output(3, "reserve_space_4", False, "optional", "all") \ .output(4, "reserve_space_5", False, "optional", "all") \ .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Default, - DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.F32_Default, + DataType.F32_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \ .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F32_5HD, DataType.F32_5HD, - DataType.F32_5HD, DataType.F32_5HD, DataType.F16_5HD, DataType.F32_5HD, + DataType.F32_5HD, DataType.F16_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \ .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, - DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, + DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \ .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, - DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, + DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \ .get_op_info()