Skip to content
Snippets Groups Projects

Add backward implementations

Merged Houssem ROUIS requested to merge hrouis/aidge_backend_cuda:Add_backward_Impl into dev
Files
31
@@ -40,9 +40,11 @@ public:
public:
void forward();
void backward();
// ~AddImpl_cuda();
private:
template <class T> void forward_(const std::vector<Tensor>& inputs, const std::vector<std::vector<int>>& inputsDims, const std::vector<std::vector<int>>& inputsStrides);
template <class T> void backward_(const Tensor& outGrad, const std::vector<std::vector<int>>& inputsDims, const std::vector<std::vector<int>>& inputsStrides);
};
namespace {
Loading