Skip to content
Snippets Groups Projects
Commit 4b179a9b authored by Maxence Naud's avatar Maxence Naud
Browse files

[Add] 'Producer_Op' tensor constructor now copies the backend of its output Tensor

parent d0165f5b
No related branches found
No related tags found
3 merge requests!105version 0.2.0,!88Basic supervised learning,!79Scheduler backward
Pipeline #40024 passed
......@@ -51,6 +51,8 @@ public:
: OperatorTensor(Type, 0, 0, 1),
Attributes_(attr<ProdAttr::Constant>(constant))
{
if (tensor->getImpl())
mImpl = Registrar<Producer_Op>::create(tensor->getImpl()->backend())(*this);
mOutputs[0] = tensor; // copy the pointer of the Tensor
}
......
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