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

Fix convertion error

parent 6faa8136
No related branches found
No related tags found
2 merge requests!50version 0.2.0,!20Vit operators
Pipeline #38724 passed
...@@ -68,7 +68,8 @@ void ReduceMeanImpl_cpu_forward_kernel(const typename ReduceMean_Op<DIM>::Attrs& ...@@ -68,7 +68,8 @@ void ReduceMeanImpl_cpu_forward_kernel(const typename ReduceMean_Op<DIM>::Attrs&
const I* inputAccumulation = input; const I* inputAccumulation = input;
I* outputAccumulation = nullptr; I* outputAccumulation = nullptr;
for (const std::size_t& a : std::get<0>(attrs)) { for (const auto& axisInt : std::get<0>(attrs)) {
const std::size_t a = static_cast<std::size_t>(axisInt);
outputElements /= inputDims[a]; outputElements /= inputDims[a];
outputAccumulation = new I[outputElements]; outputAccumulation = new I[outputElements];
const std::size_t dim_i = inputDims[a]; const std::size_t dim_i = inputDims[a];
......
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