• rocking's avatar
    Pool3d fwd (#697) · 76ec0089
    rocking authored
    * Expand the base class of pool2d, prepare to share base class with pool3d
    
    * Add pool3d device op
    
    * Add pool3d f16 example
    
    * Refactor the base class. implement generic pooling in the future
    
    * clang format
    
    * get original index in max pooling
    
    * Add outputindex to base class
    
    * Fix dimension
    
    * Add pooling instance
    
    * Use indexType instead
    
    * Remove useless header
    
    * Extract IndexDataType to template
    
    * Extract pooling reference code
    
    * clang format
    
    * clang format
    
    * Fix typo
    
    * Add tensor stride
    
    * Add missing header
    
    * Add index stride and output stride
    
    * Refine naming
    
    * Add type to base class
    
    * Rename file
    
    * Use proper size
    
    * Fix typo
    
    * Refine naming
    
    * Modify the argument into vector.
    
    * Add max pool profiler
    
    * Refine naming
    
    * Support f32 pool
    
    * Fix typo
    
    * Add avg pool2d fwd in profiler
    
    * clang format
    
    * Rename AccDatatype to ComputeDatatype
    
    * Fix init
    
    * test pool
    
    * Extract variable
    
    * Add client example
    
    * Check the pooling dim
    
    * clang format
    
    * Connect argv and arg_parser
    
    * Add found check
    
    * Remove useless header
    
    * Refine naming
    
    * Adjust the order of device_pool_fwd
    76ec0089
CMakeLists.txt 4.37 KB