• Shucai Xiao's avatar
    Improve operators for onnxruntime (#405) · 992666e6
    Shucai Xiao authored
    
    
    * improve unsqueeze to support negative axis and parsing scalar
    
    * clang format
    
    * add a test example for the negative axis of unsqueeze
    
    * improve the squeeze operator to support negative axis
    
    * clang format
    
    * fixed a small bug in the lrn implementation
    
    * clang format
    
    * support negative axis in argmax and argmin
    
    * clang format
    
    * improve flatten to support negative axis
    
    * clang format
    
    * change softmax/logsoftmax to support negative axis
    
    * clang format
    
    * improve transpose by adding default perm
    
    * clang format
    
    * add one more dimens for tensor size
    
    * add one more dimens for tensor size
    
    * disable conv ops fusion for non-symmetric cases
    
    * clang format
    
    * fixed review comments
    
    * move computing axis from the device function to the compute function
    
    * clang format
    
    * move computing axis from device function to the operator computing function
    
    * clang format
    Co-authored-by: default avatarmvermeulen <5479696+mvermeulen@users.noreply.github.com>
    992666e6
softmax.cpp 782 Bytes