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

[Upd] forwa Concat after removal of 'NbInputs' attribute

parent 3efe6acc
No related branches found
No related tags found
2 merge requests!22Update operators implementation,!16Draft: Tiling
Checking pipeline status
...@@ -42,11 +42,11 @@ void ConcatImpl_cpu_forward_kernel(const Concat_Op::Attrs& attrs, ...@@ -42,11 +42,11 @@ void ConcatImpl_cpu_forward_kernel(const Concat_Op::Attrs& attrs,
DimSize_t outputAxisValue = std::accumulate(concatAxisValues.begin(), concatAxisValues.end(), 0); DimSize_t outputAxisValue = std::accumulate(concatAxisValues.begin(), concatAxisValues.end(), 0);
DimSize_t prodDimLower = 1; DimSize_t prodDimLower = 1;
for (DimIdx_t i = 0; i < std::get<1>(attrs); ++i) { for (DimIdx_t i = 0; i < std::get<0>(attrs); ++i) {
prodDimLower *= dimsFirstInput[i]; prodDimLower *= dimsFirstInput[i];
} }
DimSize_t prodDimHigher = 1; DimSize_t prodDimHigher = 1;
for (DimIdx_t i = std::get<1>(attrs) + 1; static_cast<std::size_t>(i) < dimsFirstInput.size(); for (DimIdx_t i = std::get<0>(attrs) + 1; static_cast<std::size_t>(i) < dimsFirstInput.size();
++i) { ++i) {
prodDimHigher *= dimsFirstInput[i]; prodDimHigher *= dimsFirstInput[i];
} }
......
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