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

fix OperatorImpl call

parent 75c81956
No related branches found
No related tags found
No related merge requests found
Pipeline #42527 failed
...@@ -36,7 +36,7 @@ private: ...@@ -36,7 +36,7 @@ private:
std::shared_ptr<Tensor> mInputFallback; std::shared_ptr<Tensor> mInputFallback;
public: 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) { static std::unique_ptr<AvgPoolingImpl_cuda> create(const AvgPooling_Op<2> &op) {
return std::make_unique<AvgPoolingImpl_cuda>(op); return std::make_unique<AvgPoolingImpl_cuda>(op);
......
...@@ -44,7 +44,7 @@ private: ...@@ -44,7 +44,7 @@ private:
std::shared_ptr<Tensor> mInput2Fallback; std::shared_ptr<Tensor> mInput2Fallback;
public: 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) { static std::unique_ptr<ConvImpl_cuda> create(const Conv_Op<DIM> &op) {
return std::make_unique<ConvImpl_cuda>(op); return std::make_unique<ConvImpl_cuda>(op);
......
...@@ -38,7 +38,7 @@ private: ...@@ -38,7 +38,7 @@ private:
public: 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) { static std::unique_ptr<FCImpl_cuda> create(const FC_Op &op) {
return std::make_unique<FCImpl_cuda>(op); return std::make_unique<FCImpl_cuda>(op);
......
...@@ -36,7 +36,7 @@ private: ...@@ -36,7 +36,7 @@ private:
std::shared_ptr<Tensor> mInputFallback; std::shared_ptr<Tensor> mInputFallback;
public: 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) { static std::unique_ptr<MaxPoolingImpl_cuda> create(const MaxPooling_Op<2> &op) {
return std::make_unique<MaxPoolingImpl_cuda>(op); return std::make_unique<MaxPoolingImpl_cuda>(op);
......
...@@ -38,7 +38,7 @@ private: ...@@ -38,7 +38,7 @@ private:
std::shared_ptr<Tensor> mInputFallback; std::shared_ptr<Tensor> mInputFallback;
public: 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) { static std::unique_ptr<ReLUImpl_cuda> create(const ReLU_Op &op) {
return std::make_unique<ReLUImpl_cuda>(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