dfdx icon indicating copy to clipboard operation
dfdx copied to clipboard

Add tensor_ops::max_pool/avg_pool and nn::MaxPool/AvgPool

Open coreylowman opened this issue 1 year ago • 0 comments

This will likely require it's own device function in the future, as I suspect this will be implement similarly to conv_forward where it's just looping over the inputs. Can this re-use conv_forward/backward?

Also, should there just be a single pool method that takes an enum { MaxPool, AvgPool, ... }?

coreylowman avatar Aug 21 '22 15:08 coreylowman