We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 76de138 commit 8649fddCopy full SHA for 8649fdd
1 file changed
src/subgraph/static-reduce.c
@@ -134,6 +134,14 @@ static enum xnn_status reshape_reduce_operator(
134
reduction_axes[i] = 0 <= opdata->reduction_axes[i]
135
? opdata->reduction_axes[i]
136
: input_num_dims + opdata->reduction_axes[i];
137
+ if (reduction_axes[i] < 0 || reduction_axes[i] >= (int64_t) input_num_dims) {
138
+ xnn_log_error(
139
+ "failed to reshape Reduce operator with input ID #%" PRIu32
140
+ ": reduction axis #%d (%" PRId64
141
+ ") is out of bounds for a %zu-dimensional input",
142
+ input_id, i, reduction_axes[i], input_num_dims);
143
+ return xnn_status_invalid_parameter;
144
+ }
145
}
146
size_t input_dims[XNN_MAX_TENSOR_DIMS];
147
memcpy(input_dims, input_value->shape.dim, input_num_dims * sizeof(size_t));
0 commit comments