Searched refs:spatialBlockReduceX (Results 1 – 1 of 1) sorted by relevance
232 T spatialBlockReduceX(T *shared, T val) { in spatialBlockReduceX() function281 max_input = spatialBlockReduceX<accscalar_t, Max>(sdata,max_input); in cunn_SpatialSoftMaxForward()287 sum = spatialBlockReduceX<accscalar_t, Add>(sdata, sum); in cunn_SpatialSoftMaxForward()331 sum = spatialBlockReduceX<accscalar_t, Add>(sdata, sum); in cunn_SpatialSoftMaxBackward()