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

fix ReduceMean.hpp initializer_list as vector

parent cfd265eb
No related branches found
No related tags found
No related merge requests found
...@@ -96,7 +96,7 @@ class ReduceMean_Op : public OperatorTensor, ...@@ -96,7 +96,7 @@ class ReduceMean_Op : public OperatorTensor,
outDims.erase(outDims.begin() + static_cast<std::size_t>(*it)); outDims.erase(outDims.begin() + static_cast<std::size_t>(*it));
} }
mOutputs[0]->resize((outDims.size()>0) ? outDims : {1}); mOutputs[0]->resize((outDims.size()>0) ? outDims : std::vector<DimSize_t>({1}));
} }
} }
......
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