Skip to content
Snippets Groups Projects
Commit 2ff2cd3c authored by Houssem ROUIS's avatar Houssem ROUIS Committed by Cyril Moineau
Browse files

fix init max value with lowest numeric limit for argmax

parent d8519488
No related branches found
No related tags found
No related merge requests found
Pipeline #82477 failed
......@@ -51,7 +51,7 @@ void ArgMaxImpl_cpu_forward_kernel(std::int32_t axis_,
for (std::size_t post = 0; post < stride_post; ++post) {
const std::size_t idx_i = pre * dim_i * stride_post + post;
const std::size_t idx_o = pre * stride_post + post;
I max = std::numeric_limits<I>::min();
I max = std::numeric_limits<I>::lowest();
for (std::size_t i = 0; i < dim_i; ++i) {
I curr_value = input[idx_i + i*stride_post];
if (select_last_index) {
......
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