diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorLayerNormalization.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorLayerNormalization.cpp index 30047e750c..9b366649de 100644 --- a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorLayerNormalization.cpp +++ b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorLayerNormalization.cpp @@ -30,7 +30,7 @@ public: int32_t onnxAxis = kernelCreationContext.GetOptionalAttribute(AttrName::Axis, -1); uint32_t inputDimCount = kernelCreationContext.GetTensorShapeDescription().GetInputTensorDimensionCount(0); onnxAxis = OperatorHelper::HandleNegativeAxis(onnxAxis, inputDimCount); - std::vector onnxAxes(inputDimCount - onnxAxis); + std::vector onnxAxes(static_cast(inputDimCount) - static_cast(onnxAxis)); std::iota(onnxAxes.begin(), onnxAxes.end(), onnxAxis); assert(m_inputTensorDescs.size() == 3); diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorSkipLayerNormalization.cpp b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorSkipLayerNormalization.cpp index 5c6e98c17a..5f5a93282e 100644 --- a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorSkipLayerNormalization.cpp +++ b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorSkipLayerNormalization.cpp @@ -27,7 +27,7 @@ public: int32_t onnxAxis = kernelCreationContext.GetOptionalAttribute(AttrName::Axis, -1); uint32_t inputDimCount = kernelCreationContext.GetTensorShapeDescription().GetInputTensorDimensionCount(0); onnxAxis = OperatorHelper::HandleNegativeAxis(onnxAxis, inputDimCount); - std::vector onnxAxes(inputDimCount - onnxAxis); + std::vector onnxAxes(static_cast(inputDimCount) - static_cast(onnxAxis)); std::iota(onnxAxes.begin(), onnxAxes.end(), onnxAxis); assert(m_inputTensorDescs.size() == 5);