Skip to content
Snippets Groups Projects
Commit 3f6ee54d authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

fix OperatorImpl call

parent 75c81956
Branches lenet_operators
No related tags found
No related merge requests found
Pipeline #42527 failed
......@@ -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);
......
......@@ -44,7 +44,7 @@ private:
std::shared_ptr<Tensor> mInput2Fallback;
public:
ConvImpl_cuda(const Conv_Op<DIM> &op) : OperatorImpl(op) {}
ConvImpl_cuda(const Conv_Op<DIM> &op) : OperatorImpl(op, "cuda") {}
static std::unique_ptr<ConvImpl_cuda> create(const Conv_Op<DIM> &op) {
return std::make_unique<ConvImpl_cuda>(op);
......
......@@ -38,7 +38,7 @@ private:
public:
FCImpl_cuda(const FC_Op &op) : OperatorImpl(op) {}
FCImpl_cuda(const FC_Op &op) : OperatorImpl(op, "cuda") {}
static std::unique_ptr<FCImpl_cuda> create(const FC_Op &op) {
return std::make_unique<FCImpl_cuda>(op);
......
......@@ -36,7 +36,7 @@ private:
std::shared_ptr<Tensor> mInputFallback;
public:
MaxPoolingImpl_cuda(const MaxPooling_Op<DIM> &op) : OperatorImpl(op) {}
MaxPoolingImpl_cuda(const MaxPooling_Op<DIM> &op) : OperatorImpl(op, "cuda") {}
static std::unique_ptr<MaxPoolingImpl_cuda> create(const MaxPooling_Op<2> &op) {
return std::make_unique<MaxPoolingImpl_cuda>(op);
......
......@@ -38,7 +38,7 @@ private:
std::shared_ptr<Tensor> mInputFallback;
public:
ReLUImpl_cuda(const ReLU_Op &op) : OperatorImpl(op) {}
ReLUImpl_cuda(const ReLU_Op &op) : OperatorImpl(op, "cuda") {}
static std::unique_ptr<ReLUImpl_cuda> create(const ReLU_Op &op) {
return std::make_unique<ReLUImpl_cuda>(op);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment