Add backward() for implemented operators on CUDA
Add the backward() implementation for currently supported operators on backend_cuda:
- Add
- AvgPool
- BatchNorm
- FC
- GlobalAvgPool
- MaxPool
- Pad
- Sub
- Reshape
Edited by Houssem ROUIS
Add the backward() implementation for currently supported operators on backend_cuda:
Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent