[onert] Fix the missed nullptr check (#11515)

This commit fixes the missed nullptr check in train::FullConnectedLayer ONE-DCO-1.0-Signed-off-by: ragmani <[email protected]>
Samsung · Sep 11, 2023 · e13bdc6 · e13bdc6
1 parent e2a84ef
commit e13bdc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtime/onert/backend/train/ops/FullyConnectedLayer.cc b/runtime/onert/backend/train/ops/FullyConnectedLayer.cc
@@ -117,7 +117,7 @@ void FullyConnectedLayer::backward()
     case OperandType::FLOAT32:
     {
       assert(data_type == _grad_weights->data_type());
-      assert(data_type == _grad_bias->data_type());
+      assert(_grad_bias == nullptr || data_type == _grad_bias->data_type());
       backwardFloat32();
       break;
     }