Updated init parameter to be tensor vector
This commit is contained in:
parent
90ea083cba
commit
8aa7843f0e
7 changed files with 108 additions and 16 deletions
|
|
@ -27,7 +27,7 @@ class OpCreateTensor : public OpBase
|
|||
|
||||
~OpCreateTensor();
|
||||
|
||||
void init(std::shared_ptr<Tensor> tensor, ...) override;
|
||||
void init(std::vector<std::shared_ptr<Tensor>> tensors) override;
|
||||
|
||||
void record() override;
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue