[train] Revise maxPool2D in training

This commit change casting to static cast. ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]
Samsung · Oct 7, 2024 · 0366ecc · 0366ecc
1 parent c82f8cb
commit 0366ecc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/runtime/onert/backend/train/ops/PoolLayer.cc b/runtime/onert/backend/train/ops/PoolLayer.cc
@@ -58,8 +58,10 @@ class MaxPool2D final : public TrainingKernelRegistry
       _op_params.stride_width = strideWidth;
       _op_params.filter_height = kernelHeight;
       _op_params.filter_width = kernelWidth;
-      _op_params.padding_values.height = (int8_t)paddingTop;
-      _op_params.padding_values.width = (int8_t)paddingLeft;
+      assert(paddingTop < (1 << 8));
+      assert(paddingLeft < (1 << 8));
+      _op_params.padding_values.height = static_cast<uint8_t>(paddingTop);
+      _op_params.padding_values.width = static_cast<uint8_t>(paddingLeft);
       CalculateActivationRange<float>(activation, &_op_params.float_activation_min,
                                       &_op_params.float_activation_max);
     }