Describe the bug
The new specifications of ONNX add axes as an attribute, however in the LoweringOptimizationPasses.py it is indexed as axis, which results in an error when the graph contains a ReduceMean node.
To Reproduce
Run any test with a ReduceMean operation.
Expected behavior
Generate network successfully once _remove_only_singleton_reduce_mean function is fixed.
Additional context
I will open a PR for this, leaving this as mental note.
Describe the bug
The new specifications of ONNX add
axesas an attribute, however in theLoweringOptimizationPasses.pyit is indexed asaxis, which results in an error when the graph contains a ReduceMean node.To Reproduce
Run any test with a ReduceMean operation.
Expected behavior
Generate network successfully once
_remove_only_singleton_reduce_meanfunction is fixed.Additional context
I will open a PR for this, leaving this as mental note.