Searched refs:BlockReduceMax (Results 1 – 2 of 2) sorted by relevance
104 auto final_val = at::native::cuda_utils::BlockReduceMax(val, s_vals); in operator ()()126 T final_val = at::native::cuda_utils::BlockReduceMax(val, vals); in lpmax_cleanup()310 : at::native::cuda_utils::BlockReduceMax(val, s_vals); in operator ()()344 : at::native::cuda_utils::BlockReduceMax(val, vals); in lpnorm_cleanup()
96 __inline__ __device__ T BlockReduceMax(T val, T* shared) { in BlockReduceMax() function