Updated init parameter to be tensor vector

2020-08-20 05:27:42 +01:00 · 2020-08-20 05:27:42 +01:00 · 8aa7843f0e
commit 8aa7843f0e
parent 90ea083cba
7 changed files with 108 additions and 16 deletions
--- a/src/OpBase.hpp
+++ b/src/OpBase.hpp
@ -37,7 +37,7 @@ class OpBase
        SPDLOG_DEBUG("Compute OpBase destructor started");
    }

-    virtual void init(std::shared_ptr<Tensor> tensor, ...) {
+    virtual void init(std::vector<std::shared_ptr<Tensor>> tensors) {
        SPDLOG_DEBUG("Kompute OpBase init called");
    }