diff options
Diffstat (limited to 'src/cpu/kernels')
-rw-r--r-- | src/cpu/kernels/conv3d/neon/list.h | 2 | ||||
-rw-r--r-- | src/cpu/kernels/conv3d/neon/quantized.h | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/src/cpu/kernels/conv3d/neon/list.h b/src/cpu/kernels/conv3d/neon/list.h index 3e2db664d7..3bfa124dc3 100644 --- a/src/cpu/kernels/conv3d/neon/list.h +++ b/src/cpu/kernels/conv3d/neon/list.h @@ -150,7 +150,7 @@ void directconv3d_float_neon_ndhwc(const ITensor *src0, const ITensor *src1, con { const auto src_vec = wrapper::vloadq(in_ptr_mover); //Load Cin weights - for(unsigned int k = 0; k < num_elems_read_per_iteration; ++k, weights_ptr_mover += index_c_out_end) + for(int k = 0; k < num_elems_read_per_iteration; ++k, weights_ptr_mover += index_c_out_end) { w_vec = wrapper::vsetlane(*weights_ptr_mover, w_vec, k); } diff --git a/src/cpu/kernels/conv3d/neon/quantized.h b/src/cpu/kernels/conv3d/neon/quantized.h index cdbc45e207..a8165b4944 100644 --- a/src/cpu/kernels/conv3d/neon/quantized.h +++ b/src/cpu/kernels/conv3d/neon/quantized.h @@ -171,7 +171,7 @@ void directconv3d_quantized_neon_ndhwc(const ITensor *src0, const ITensor *src1, { const auto src_vec = wrapper::vloadq(in_ptr_mover); //Load Cin weights - for(unsigned int k = 0; k < num_elems_read_per_iteration; ++k, weights_ptr_mover += index_c_out_end) + for(int k = 0; k < num_elems_read_per_iteration; ++k, weights_ptr_mover += index_c_out_end) { w_vec = wrapper::vsetlane(*weights_ptr_mover, w_vec, k); } |