From bd01a358f8dfa18cfb3c1769731582251ce0e2c6 Mon Sep 17 00:00:00 2001
From: liuxiao93 <liuxiao93@huawei.com>
Date: Thu, 15 Apr 2021 17:01:05 +0800
Subject: [PATCH] fix output of BiasAddGrad.

---
 mindspore/ops/operations/_grad_ops.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/ops/operations/_grad_ops.py b/mindspore/ops/operations/_grad_ops.py
index 010b637f39..fb9d92f355 100644
--- a/mindspore/ops/operations/_grad_ops.py
+++ b/mindspore/ops/operations/_grad_ops.py
@@ -232,6 +232,8 @@ class BiasAddGrad(PrimitiveWithInfer):
         self.format = validator.check_string(data_format, ['NCHW', 'NHWC', 'NCDHW'], 'format', self.name)
         if context.get_context("device_target") != "GPU" and self.format == "NHWC":
             raise ValueError("NHWC format only support in GPU target.")
+        if self.format == "NCDHW":
+            self.format = "NCHW"
         self.add_prim_attr('data_format', self.format)
 
     def infer_shape(self, d_output):