Where is the gpu implementation of nonzero kernel

ScXfjiang (scxfjiang) October 10, 2019, 6:08am 1

Where is the gpu implementation of nonzero kernel, I can’t find it, thanks!

tom (Thomas V) October 10, 2019, 6:33am 2

It’s in THC (legacy torch) as nonzero in /aten/src/THC/generic/THCTensorMath.cu.

Best regards

Thomas