diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/bias_add_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/bias_add_cpu_kernel.cc
index 015bea0ec43..c3ac19dbd6c 100644
--- a/mindspore/ccsrc/plugin/device/cpu/kernel/bias_add_cpu_kernel.cc
+++ b/mindspore/ccsrc/plugin/device/cpu/kernel/bias_add_cpu_kernel.cc
@@ -45,6 +45,12 @@ int BiasAddCpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const std:
   if (ret != KRET_OK) {
     return ret;
   }
+  for (const auto &input : inputs) {
+    auto input_shape = input->GetShapeVector();
+    if (!IsValidShape(input_shape)) {
+      return KRET_UNKNOWN_SHAPE;
+    }
+  }
   input_shape_ = Convert2SizeTClipNeg(inputs[kIndex0]->GetShapeVector());
   bias_shape_ = Convert2SizeTClipNeg(inputs[kIndex1]->GetShapeVector());
   data_shape_ = input_shape_.size();
@@ -143,6 +149,7 @@ const std::vector<std::pair<KernelAttr, BiasAddCpuKernelMod::KernelRunFunc>> &Bi
     MakeKernelFunc<int8_t>(kNumberTypeInt8),
     MakeKernelFunc<int16_t>(kNumberTypeInt16),
     MakeKernelFunc<int64_t>(kNumberTypeInt32),
+    MakeKernelFunc<int64_t>(kNumberTypeInt64),
     MakeKernelFunc<uint8_t>(kNumberTypeUInt8),
     MakeKernelFunc<uint16_t>(kNumberTypeUInt16),
     MakeKernelFunc<uint32_t>(kNumberTypeUInt32),
diff --git a/mindspore/ccsrc/plugin/device/gpu/kernel/nn/bias_add_gpu_kernel.cc b/mindspore/ccsrc/plugin/device/gpu/kernel/nn/bias_add_gpu_kernel.cc
index 71f83516a3e..e0cf32afbe2 100644
--- a/mindspore/ccsrc/plugin/device/gpu/kernel/nn/bias_add_gpu_kernel.cc
+++ b/mindspore/ccsrc/plugin/device/gpu/kernel/nn/bias_add_gpu_kernel.cc
@@ -45,6 +45,12 @@ int BiasAddGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const std:
   if (auto ret = KernelMod::Resize(base_operator, inputs, outputs, inputsOnHost); ret != KRET_OK) {
     return ret;
   }
+  for (const auto &input : inputs) {
+    auto input_shape = input->GetShapeVector();
+    if (!IsValidShape(input_shape)) {
+      return KRET_UNKNOWN_SHAPE;
+    }
+  }
   auto x_shape = LongVecToSizeVec(inputs[kIndex0]->GetShapeVector());
   auto num_dims = x_shape.size();
   is_null_input_ = CHECK_SHAPE_NULL(x_shape, kernel_name_, "input_x");