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

[Fix] Attributes first index result in Conv

parent 3fb21e1a
No related branches found
No related tags found
No related merge requests found
...@@ -171,8 +171,8 @@ public: ...@@ -171,8 +171,8 @@ public:
// Result // Result
std::vector<std::pair<std::size_t, std::vector<DimSize_t>>> res; std::vector<std::pair<std::size_t, std::vector<DimSize_t>>> res;
res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[0]->getIdx(inputIdxDims), inputDims)); res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[0]->getIdx(inputIdxDims), inputDims));
res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(weightIdxDims, weightDims)); res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[1]->getIdx(weightIdxDims), weightDims));
res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(biasIdxDims, biasDims)); res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[2]->getIdx(biasIdxDims), biasDims));
return res; return res;
} }
AIDGE_THROW_OR_ABORT(std::runtime_error, "Given outputDim out of range or output dim not forwarded yet."); AIDGE_THROW_OR_ABORT(std::runtime_error, "Given outputDim out of range or output dim not forwarded yet.");
......
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