• Sang Ik Lee's avatar
    TopK (w/ArgMax, ArgMin python wrapper) (#1560) · 3548772b
    Sang Ik Lee authored
    * Implement TopK.
    
    * Update python wrappers for TopK, ArgMin and ArgMax.
    
    * Address some reviewer comments.
    
    * Add type property check tests for TopK.
    Set correct TopK behavior for K==0.
    
    * TopK: Add 1d and 3d unit tests.
    
    * Address more reviewer comments.
    
    * Apply code style.
    3548772b
Name
Last commit
Last update
..
files Yet another serialization option (#619)
models [ONNX] Numpy style binary broadcasting (#1549)
ref_generators Change license header to use single-line comment (#1508)
util Validate/infer types as a virtual function (#1463)
CMakeLists.txt [ONNXIFI] implement onnxGetBackendIDs() interface function (#1546)
algebraic_simplification.cpp Change license header to use single-line comment (#1508)
all_close_f.cpp Change license header to use single-line comment (#1508)
assertion.cpp Change license header to use single-line comment (#1508)
autodiff.in.cpp Simplify result copy elimination (#1345)
backend_api.cpp Change license header to use single-line comment (#1508)
backend_debug_api.cpp Change license header to use single-line comment (#1508)
backend_performance.cpp Change license header to use single-line comment (#1508)
backend_test.in.cpp TopK (w/ArgMax, ArgMin python wrapper) (#1560)
build_graph.cpp Validate/infer types as a virtual function (#1463)
builder.cpp Change license header to use single-line comment (#1508)
builder_autobroadcast.cpp Change license header to use single-line comment (#1508)
constant_folding.cpp Change license header to use single-line comment (#1508)
convolution_test.in.cpp Change license header to use single-line comment (#1508)
copy.cpp Change license header to use single-line comment (#1508)
core_fusion.cpp Change license header to use single-line comment (#1508)
cpio.cpp Change license header to use single-line comment (#1508)
cpu_fusion.cpp Change license header to use single-line comment (#1508)
cpu_test.cpp Change license header to use single-line comment (#1508)
cse.cpp Change license header to use single-line comment (#1508)
cudnn.cpp Change license header to use single-line comment (#1508)
distributed.cpp Loading commit data...
element_type.cpp Loading commit data...
file_util.cpp Loading commit data...
gpu_test.cpp Loading commit data...
graph_partition.cpp Loading commit data...
includes.cpp Loading commit data...
inliner.cpp Loading commit data...
input_output_assign.cpp Loading commit data...
main.cpp Loading commit data...
mkldnn.cpp Loading commit data...
nop_elimination.cpp Loading commit data...
onnx_import.cpp Loading commit data...
onnxifi.cpp Loading commit data...
op.cpp Loading commit data...
pass_liveness.cpp Loading commit data...
pass_manager.cpp Loading commit data...
pass_memory_layout.cpp Loading commit data...
pattern.cpp Loading commit data...
reshape_elimination.cpp Loading commit data...
serialize.cpp Loading commit data...
shape.cpp Loading commit data...
tensor.cpp Loading commit data...
type_prop.cpp Loading commit data...
update_reference.sh Loading commit data...
util.cpp Loading commit data...
uuid.cpp Loading commit data...
zero_dim_tensor_elimination.cpp Loading commit data...