• Pruthvi's avatar
    Cumulative Sum (#3873) · 77a99b30
    Pruthvi authored
    * - Op defination for cummalative sum
    
    * WIP reference kernel for cummulative sum
    
    * - unit test case for default cum_sum
    - addition ctor for cumsum to accept axis as a integer insted of Node
    type
    - style fix
    
    * - add serializer support
    - fix failing unit test case
    - update Op in the interpreter dispatcher
    
    * - CPU builder and DEX support for CumSum
    
    * - implemented mapping tensor elements to corrosponding axis
    
    * - unit test for multiple dims
    - fix axis in the op defination
    - support for reference kernel to compute across all axis
    
    * - added support for exclusive and reverse modes
    - more unit test case for all modes
    
    * - codegen support for CumSum
    - disable CumSum unit test for PlaidML
    
    * -Add missing header to  codegen stream writer
    
    * fixed codegen writer
    
    * change return type of exclusive and reverse to bool
    
    * - support for dynamic shape
    - support to handle all tensor types in CPU builder
    
    * - add support for interpreter to handle different axis types
    
    * Style fix
    77a99b30
cum_sum.in.cpp 7.76 KB