Skip to content

[backend_cuda] 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