Skip to content

Commit 3bedf32

Browse files
Update onnxruntime/core/providers/cpu/math/einsum_utils/einsum_typed_compute_processor.cc
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 50582ac commit 3bedf32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxruntime/core/providers/cpu/math/einsum_utils/einsum_typed_compute_processor.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ Status EinsumTypedComputeProcessor<T>::Run() {
370370
TensorShapeVector reduced_dims; // All dims of the input that are reduced using the `ReduceSum` op
371371
reduced_dims.reserve(onnxruntime::narrow<size_t>(num_subscript_labels)); // num_subscript_labels is the upper bound. No harm in over-reserving
372372

373-
TensorShapeVector all_dims; // Expanded dimensions of `num_subscript_labels` {0, 1, ..., num_subscript_labels}
373+
TensorShapeVector all_dims; // All dimension indices from 0 to num_subscript_labels - 1
374374
all_dims.reserve(onnxruntime::narrow<size_t>(num_subscript_labels)); // num_subscript_labels is the number of elements
375375

376376
for (size_t i = 0; i < onnxruntime::narrow<size_t>(num_subscript_labels); ++i) {

0 commit comments

Comments
 (0)