diff --git a/tensorflow/core/kernels/mkl_fused_batch_norm_op.cc b/tensorflow/core/kernels/mkl_fused_batch_norm_op.cc index 7c8919929f8da5..4fbacc13e5a438 100644 --- a/tensorflow/core/kernels/mkl_fused_batch_norm_op.cc +++ b/tensorflow/core/kernels/mkl_fused_batch_norm_op.cc @@ -518,6 +518,8 @@ class MklFusedBatchNormOp : public OpKernel { errors::InvalidArgument("Invalid data format")); OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); depth_ = 0; + mean_values_ = nullptr; + variance_values_ = nullptr; } void Compute(OpKernelContext* context) override {