Home
last modified time | relevance | path

Searched refs:inclusiveBinaryPrefixScan (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
DScanUtils.cuh15 __device__ void inclusiveBinaryPrefixScan(T* smem, bool in, T* out, BinaryFunction binop) { in inclusiveBinaryPrefixScan() function
65 inclusiveBinaryPrefixScan<T, false, BinaryFunction>(smem, in, out, binop); in exclusiveBinaryPrefixScan()