From 1e66b7748d155d5a9da233e1408d69cf4231b49b Mon Sep 17 00:00:00 2001 From: Jeff Bloomfield Date: Fri, 24 Jan 2020 16:24:24 -0800 Subject: [PATCH] Fix slice conformance failures --- .../core/providers/dml/OperatorAuthorHelper/OperatorHelper.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.h b/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.h index c944c1b59b27e..aa8486117fa97 100644 --- a/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.h +++ b/onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.h @@ -584,6 +584,9 @@ class SliceHelperBase ReadIndexTensors(operatorInfo, starts, ends, axes, steps); } } + + const uint32_t dimCount = gsl::narrow_cast(inputDimensions.size()); + HandleNegativeAxes(/*inout*/ axes, dimCount); ML_CHECK_VALID_ARGUMENT(starts.size() == ends.size(), "'starts' must equal 'ends' in size."); ML_CHECK_VALID_ARGUMENT(axes.empty() || starts.size() == axes.size(), "'axes' must equal 'starts' in size, or 'axes' must be empty.");