• Michał Karzyński's avatar
    [ONNX] Support for Gather op (#2772) · 0a5c2b81
    Michał Karzyński authored
    * Temp.
    
    * Put all the dummy files.
    
    * Remove some compile errors.
    
    * WIP: Add gather and gather_nd kernels.
    
    * Temp save.
    
    * Update comments for gather.
    
    * Implement reference gather.
    
    * Validate and infer shape.
    
    * Style.
    
    * Fix compile issues.
    
    * Add serializer support.
    
    * Fix interpreter compilation issues.
    
    * WIP: Add UT
    
    * WIP: Add UT
    
    * gather_nd UT passing.
    
    * Fix gather with no axis.
    
    * Fix gather issue.
    
    * Update unit_test.manifest for backends and add gather, gather_nd  support for generic cpu.
    
    * Add type_prop tests.
    
    * Add CPU builders.
    
    * Fix codegen.
    
    * Make some UT numbers more readable.
    
    * Style.
    
    * Update Copyright Year
    
    * Update Copyright Year
    
    * Fix Typo.
    
    * Remove unused variable.
    
    * fix nv gpu build error
    
    * Fix intel gpu compilation.
    
    * Add basic docstring.
    
    * Allow 1D indices for gather_nd.
    
    * Allow scalar indices for gather.
    
    * Update unit_test manifest files.
    
    * Style.
    
    * Add indices element type check and add failing type_prop checks.
    
    * Update docstring.
    
    * Fix incorrect test names in unit_test.manifest
    
    * [ONNX] Support for Gather op
    
    * Remove unneeded broadcast
    
    * Remove unused include
    
    * Set correct default value for axis
    0a5c2b81
Name
Last commit
Last update
..
ngraph Loading commit data...
resource Loading commit data...
tools Loading commit data...
CMakeLists.txt Loading commit data...