diff --git a/onnxruntime/core/graph/dml_ops/dml_defs.cc b/onnxruntime/core/graph/dml_ops/dml_defs.cc
index 3e67c9e3a9..58ebe84834 100644
--- a/onnxruntime/core/graph/dml_ops/dml_defs.cc
+++ b/onnxruntime/core/graph/dml_ops/dml_defs.cc
@@ -108,6 +108,7 @@ void RegisterDmlSchemas() {
     .Attr("spatial", "", AttributeProto::INT, static_cast<int64_t>(1))
     .Attr("epsilon", "", AttributeProto::FLOAT, 1e-5f)
     .Attr("momentum", "", AttributeProto::FLOAT, 0.9f)
+    .Attr("training_mode", "", AttributeProto::INT, static_cast<int64_t>(0))
     .Input(0, "X", "", "T")
     .Input(1, "scale", "", "T")
     .Input(2, "B", "", "T")