From 2adb0f8d1d28a268ba33994958fee3aacb5f6d33 Mon Sep 17 00:00:00 2001
From: jake mathern <j.mathern@digipen.edu>
Date: Tue, 19 May 2020 15:07:33 -0700
Subject: [PATCH] Fix 5d concat operator helper

---
 .../core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp b/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
index 78c045a4e1..07de6d1988 100644
--- a/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
+++ b/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
@@ -971,7 +971,7 @@ namespace OperatorHelper
     std::vector<EdgeShapes> ConcatHelper::GetOutputShapes(const MLShapeInferenceContext& shapeInfo) const
     {
         auto outputShape = shapeInfo.GetInputTensorShape(0);
-        ML_CHECK_VALID_ARGUMENT(outputShape.size() <= NchwDimensionCount);
+        ML_CHECK_VALID_ARGUMENT(outputShape.size() <= NcdhwDimensionCount);
 
         uint32_t inputCount = shapeInfo.GetInputCount();