1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
diff --git a/onnxruntime/core/providers/cuda/reduction/reduction_functions.cc b/onnxruntime/core/providers/cuda/reduction/reduction_functions.cc
index 955df6d9a..f9fd53e15 100644
--- a/onnxruntime/core/providers/cuda/reduction/reduction_functions.cc
+++ b/onnxruntime/core/providers/cuda/reduction/reduction_functions.cc
@@ -39,7 +39,7 @@ optional<std::pair<int64_t, int64_t>> GetMinAndMaxContiguousAxes(
}
// normalize axis values and sort
- const std::vector<int64_t> axes = [&original_axes, rank]() {
+ const std::vector<int64_t> axes = [&original_axes, rank]() -> std::vector<int64_t> {
std::vector<int64_t> result(original_axes);
std::for_each(
result.begin(), result.end(),
@@ -85,7 +85,7 @@ optional<std::pair<int64_t, int64_t>> GetMinAndMaxContiguousAxes(
return std::distance(dims.begin(), before_min_axis_rit.base());
}();
- const int64_t max_axis = [&dims, &axes, &is_dim_one]() {
+ const int64_t max_axis = [&dims, &axes, &is_dim_one]() -> int64_t {
const auto& max_given_axis = axes.back();
const auto after_max_given_axis_it = dims.begin() + max_given_axis + 1;
const auto after_max_axis_it =
|