-
Mateusz Bencer authored
[SPEC][ONNX] Handle negative axis for TopK:v1, add dynamic shape support for ONNX Arg Min/Max ops (#4291) * First version * Added support to no_keep_dims * Excluded tests for PlaidML * Code review remarks introduced * Reduced TopK axis restrictions * Added assert to TopK get_axis * Added missing EOF * Style applied * Code review suggestions introduced * Disable tests for GPU * Code review remarks introduced Co-authored-by: Scott Cyphers <diyessi@users.noreply.github.com>
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
mxnet | ||
onnx | ||
paddlepaddle | ||
tensorflow | ||
conv_bias.json | ||
tf_conv_mnist_nhwc.json |