diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp index 8343cd1b2a..05dcd49dc8 100644 --- a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp +++ b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp @@ -862,7 +862,11 @@ namespace Dml { return; } - + else if (!IsCpuData()) + { + return; + } + uint32_t totalKernelInputElementCount = constExpTensor->GetTotalElementCount(); if (totalKernelInputElementCount <= 1) {