[train] fix padding_value type (#14206)

This commit fixes type of padding value match with declaration. ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]
Samsung · Oct 15, 2024 · e12da49 · e12da49
1 parent 4a319d0
commit e12da49
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/runtime/onert/backend/train/ops/PoolLayer.cc b/runtime/onert/backend/train/ops/PoolLayer.cc
@@ -134,10 +134,10 @@ class AveragePool2D final : public TrainingKernelRegistry
       _op_params.stride_width = strideWidth;
       _op_params.filter_height = kernelHeight;
       _op_params.filter_width = kernelWidth;
-      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);
+      assert(paddingTop < (1 << 15));
+      assert(paddingLeft < (1 << 15));
+      _op_params.padding_values.height = static_cast<int16_t>(paddingTop);
+      _op_params.padding_values.width = static_cast<int16_t>(paddingLeft);
       CalculateActivationRange<float>(activation, &_op_params.float_activation_min,
                                       &_op_params.float_activation_max);
     }