• Michał Karzyński's avatar
    [ONNX] Add support for ONNX 1.6 TopK (#3771) · 7617d385
    Michał Karzyński authored
    * Minor cleanup
    
    * Add support for ONNX 1.5 version of TopK
    
    * Add unit tests
    
    * Style apply
    
    * Exclude failing tests
    
    * Exclude failing tests
    
    * Add support for ONNX 1.6 TopK attribures: larges and sorted
    
    * Support for ONNX 1.6 TopK
    
    * If k_node is a Constant, recreate as constant with Shape{}
    
    * Extend `interpret_as_scalar` function
    
    * Extend `interpret_as_scalar` function
    
    * Remove merge artifact
    
    * Add doc string
    
    * Exclude failing tests
    
    * Exclude failing tests
    
    * Refactor function
    
    * Remove unnecessary template param
    
    * Use get_k function in OpSet 10 TopK
    
    * Style apply
    
    * Remove merge artifact
    
    * Add tests for `interpret_as_scalar`
    
    * Revert "Add tests for `interpret_as_scalar`"
    
    This reverts commit 8b85965acb39c75ff9e66b06ad8f64df16e1a9da.
    7617d385
top_k_opset_11_const_k_smallest.prototxt 1.34 KB