search

Iainmon / ChAI

A Chapel library for Machine Learning that supports distributed inference, automatic differentiation, and CUDA/HIP utilization.
https://iainmon.github.io/ChAI
4 stars 5 forks source link

[Feature Request] Implement `avgPool` and `adaptiveAvgPool` #9

Open ShreyasKhandekar opened 2 months ago

ShreyasKhandekar commented 2 months ago

See https://pytorch.org/docs/stable/generated/torch.nn.AvgPool1d.html#avgpool2d and https://pytorch.org/docs/stable/generated/torch.nn.AdaptiveAvgPool2d.html#torch.nn.AdaptiveAvgPool2d for reference.

ShreyasKhandekar commented 1 month ago

Jade has already implemented adaptiveAvgPool2d