IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implementation of ArgMin ONNX + TF extractors (#5126)

Browse Source 
* implement argmin extractors

* reconsidering argmax to topk

* arg ops refactoring

* rename ArgMaxToTopK

* added unittests

* update docs

* move unittest file to new folder

* conversations resolving

* revert changes with argmax.py, move argmin op to a new file

* rename ArgMaxSqueeze

* updated BOM file

* little fix

* code refactoring in ArgMaxOp, updated unittests

Co-authored-by: yegor.kruglov <ykruglov@nnlvdp-mkaglins.inn.intel.com>
This commit is contained in:
Yegor Kruglov
2021-05-06 13:41:49 +03:00
committed by GitHub
parent e3ea9bf4dd
commit abb1ca657e
12 changed files with 327 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/MO_DG/prepare_model/Supported_Frameworks_Layers.md
View File

@@ -128,6 +128,7 @@ Standard TensorFlow\* operations:
| AddV2 | No |
| AddN | No |
| ArgMax | No |
| ArgMin | No |
| Asinh | No |
| Atanh | No |
| AvgPool | No |
@@ -398,6 +399,7 @@ Standard ONNX\* operators:
| Add | No |
| Affine | No |
| ArgMax | No |
| ArgMin | No |
| Asin | No |
| Asinh | No |
| Atan | No |