diff --git a/runtime/onert/backend/train/ops/FullyConnectedLayer.cc b/runtime/onert/backend/train/ops/FullyConnectedLayer.cc index 8fdc822d22f..7981a898b15 100644 --- 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; }