saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle negative axes for reduce ops (#365)

Browse source 
* Handle negative axes for reduce ops

* negative axes are not handled in shape inference if input shape
 is not known at that time.

* nit: use HandleNegativeAxis in provider/common.h
This commit is contained in:
Bowen Bao 2019-01-24 16:40:54 -08:00 committed by Raymond Yang
parent f94fdad861
commit 644c13050b
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
onnxruntime/core/providers/cpu/reduction/reduction_ops.cc
View file

@ -2,6 +2,7 @@
// Licensed under the MIT License.
#include "core/providers/cpu/reduction/reduction_ops.h"
#include "core/providers/common.h"
#include "core/util/math_cpuonly.h"
using namespace std;
namespace onnxruntime {
@ -53,11 +54,11 @@ bool PrepareForReduce(OpKernelContext* ctx,
const Tensor& input = *input_tensor_ptr;
size_t ndim = input.Shape().GetDims().size();
for (int64_t axe : axes_) {
ORT_ENFORCE(axe >= 0 && axe < (int64_t)ndim, "Axis attribute out of range");
std::vector<int64_t> axes;
for (int64_t axis : axes_) {
axes.push_back(HandleNegativeAxis(axis, static_cast<int64_t>(ndim)));
}
std::vector<int64_t> axes = axes_;
if (axes.empty()) {
// This is the default case for non-arg kind reductions. Reduce on all dimensions.
for (size_t i = 0; i < ndim; i++)