• 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
Name
Last commit
Last update
..
autodiff Loading commit data...
builder Loading commit data...
codegen Loading commit data...
descriptor Loading commit data...
distributed Loading commit data...
frontend Loading commit data...
op Loading commit data...
pass Loading commit data...
pattern Loading commit data...
runtime Loading commit data...
state Loading commit data...
type Loading commit data...
CMakeLists.txt Loading commit data...
assertion.hpp Loading commit data...
axis_set.cpp Loading commit data...
axis_set.hpp Loading commit data...
axis_vector.cpp Loading commit data...
axis_vector.hpp Loading commit data...
check.hpp Loading commit data...
code_writer.hpp Loading commit data...
component_manager.hpp Loading commit data...
coordinate.cpp Loading commit data...
coordinate.hpp Loading commit data...
coordinate_diff.cpp Loading commit data...
coordinate_diff.hpp Loading commit data...
coordinate_transform.cpp Loading commit data...
coordinate_transform.hpp Loading commit data...
cpio.cpp Loading commit data...
cpio.hpp Loading commit data...
deprecated.hpp Loading commit data...
dimension.cpp Loading commit data...
dimension.hpp Loading commit data...
distributed.cpp Loading commit data...
distributed.hpp Loading commit data...
event_tracing.cpp Loading commit data...
event_tracing.hpp Loading commit data...
except.hpp Loading commit data...
file_util.cpp Loading commit data...
file_util.hpp Loading commit data...
function.cpp Loading commit data...
function.hpp Loading commit data...
graph_util.cpp Loading commit data...
graph_util.hpp Loading commit data...
log.cpp Loading commit data...
log.hpp Loading commit data...
ngraph.cpp Loading commit data...
ngraph.hpp Loading commit data...
ngraph_visibility.hpp Loading commit data...
node.cpp Loading commit data...
node.hpp Loading commit data...
partial_shape.cpp Loading commit data...
partial_shape.hpp Loading commit data...
placement.cpp Loading commit data...
placement.hpp Loading commit data...
provenance.cpp Loading commit data...
provenance.hpp Loading commit data...
rank.hpp Loading commit data...
serializer.cpp Loading commit data...
serializer.hpp Loading commit data...
serializer_stub.cpp Loading commit data...
shape.cpp Loading commit data...
shape.hpp Loading commit data...
shape_util.cpp Loading commit data...
shape_util.hpp Loading commit data...
slice_plan.cpp Loading commit data...
slice_plan.hpp Loading commit data...
specialize_function.cpp Loading commit data...
specialize_function.hpp Loading commit data...
strides.cpp Loading commit data...
strides.hpp Loading commit data...
type.hpp Loading commit data...
util.cpp Loading commit data...
util.hpp Loading commit data...
validation_util.cpp Loading commit data...
validation_util.hpp Loading commit data...
variant.cpp Loading commit data...
variant.hpp Loading commit data...
version.in.hpp Loading commit data...
visibility.hpp Loading commit data...