Skip to content
Snippets Groups Projects

Lenet operators

Merged Houssem ROUIS requested to merge hrouis/aidge_backend_cuda:lenet_operators into dev
1 unresolved thread
Compare and Show latest version
6 files
+ 5
141
Compare changes
  • Side-by-side
  • Inline
Files
6
@@ -36,7 +36,7 @@ private:
std::shared_ptr<Tensor> mInputFallback;
public:
AvgPoolingImpl_cuda(const AvgPooling_Op<DIM> &op) : OperatorImpl(op) {}
AvgPoolingImpl_cuda(const AvgPooling_Op<DIM> &op) : OperatorImpl(op, "cuda") {}
static std::unique_ptr<AvgPoolingImpl_cuda> create(const AvgPooling_Op<2> &op) {
return std::make_unique<AvgPoolingImpl_cuda>(op);
Loading