From fc0143fe68390748e1001eb252d263c740eb2946 Mon Sep 17 00:00:00 2001 From: Dwayne Robinson Date: Mon, 27 Jun 2022 19:41:34 -0700 Subject: [PATCH] DML EP ResNet50 opset 15 fails in ONNX checker for FusedBatchNormalization lacking training_mode attribute (#12010) FusedBatchNormalization include training_mode attribute --- onnxruntime/core/graph/dml_ops/dml_defs.cc | 1 + 1 file changed, 1 insertion(+) 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(1)) .Attr("epsilon", "", AttributeProto::FLOAT, 1e-5f) .Attr("momentum", "", AttributeProto::FLOAT, 0.9f) + .Attr("training_mode", "", AttributeProto::INT, static_cast(0)) .Input(0, "X", "", "T") .Input(1, "scale", "", "T") .Input(2, "B", "", "T")