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

Revise topk (#3819)

Browse Source 
* Add visit_attribute and node validation check

* add type_prop test for default values

* style-apply

* Update node validation check for index_element_type

* Update type_prop test for default index_element_type

* Add index_element_type attribute to TopK_1 spec
This commit is contained in:
Piotr Szmelczynski
2021-01-25 12:19:59 +01:00
committed by GitHub
parent d3f0242f58
commit 96b2ffa9ab
3 changed files with 28 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/ops/sort/TopK_1.md
View File

@@ -32,6 +32,14 @@
* **Default value**: None
* **Required**: *yes*
* *index_element_type*
* **Description**: the type of output tensor with indices
* **Range of values**: "i64" or "i32"
* **Type**: string
* **Default value**: "i32"
* **Required**: *No*
**Inputs**:
* **1**: Arbitrary tensor. Required.