• 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
..
backend Loading commit data...
files Loading commit data...
mlir Loading commit data...
models Loading commit data...
onnx Loading commit data...
opset_pass Loading commit data...
ref_generators Loading commit data...
type_prop Loading commit data...
util Loading commit data...
CMakeLists.txt Loading commit data...
algebraic_simplification.cpp Loading commit data...
aligned_buffer.cpp Loading commit data...
all_close_f.cpp Loading commit data...
assertion.cpp Loading commit data...
backend_api.cpp Loading commit data...
backend_debug_api.cpp Loading commit data...
backend_performance.cpp Loading commit data...
bfloat16.cpp Loading commit data...
build_graph.cpp Loading commit data...
builder.cpp Loading commit data...
builder_autobroadcast.cpp Loading commit data...
builder_quantization.cpp Loading commit data...
check.cpp Loading commit data...
concat_fusion.cpp Loading commit data...
constant_folding.cpp Loading commit data...
control_dependencies.cpp Loading commit data...
coordinate.cpp Loading commit data...
copy.cpp Loading commit data...
core.cpp Loading commit data...
core_fusion.cpp Loading commit data...
cpio.cpp Loading commit data...
cpu_codegen.cpp Loading commit data...
cpu_debug_tracer.cpp Loading commit data...
cpu_debugger.cpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cpu_test.cpp Loading commit data...
cse.cpp Loading commit data...
dyn_elimination.cpp Loading commit data...
element_type.cpp Loading commit data...
event_tracing.cpp Loading commit data...
file_util.cpp Loading commit data...
float16.cpp Loading commit data...
halide.cpp Loading commit data...
includes.cpp Loading commit data...
input_output_assign.cpp Loading commit data...
main.cpp Loading commit data...
misc.cpp Loading commit data...
misc.hpp Loading commit data...
mkldnn.cpp Loading commit data...
ngraph_api.cpp Loading commit data...
node_input_output.cpp Loading commit data...
nop_elimination.cpp Loading commit data...
op.cpp Loading commit data...
partial_shape.cpp Loading commit data...
pass.cpp Loading commit data...
pass_liveness.cpp Loading commit data...
pass_manager.cpp Loading commit data...
pass_memory_layout.cpp Loading commit data...
pass_shape_relevance.cpp Loading commit data...
pattern.cpp Loading commit data...
provenance.cpp Loading commit data...
replace_node.cpp Loading commit data...
reshape_elimination.cpp Loading commit data...
reshape_sinking.cpp Loading commit data...
serialize.cpp Loading commit data...
shape.cpp Loading commit data...
specialize_function.cpp Loading commit data...
tensor.cpp Loading commit data...
tools.cpp Loading commit data...
type_prop_benchmark.cpp Loading commit data...
type_prop_layers.cpp Loading commit data...
update_convolution_reference.sh Loading commit data...
update_dyn_replace_slice_reference.sh Loading commit data...
update_dyn_slice_reference.sh Loading commit data...
util.cpp Loading commit data...
zero_dim_tensor_elimination.cpp Loading commit data...