add static_cast

Asthestarsfalll · Sep 28, 2023 · 743e614 · 743e614
1 parent a26aa2c
commit 743e614
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/frontends/paddle/src/op/partial_concat.cpp b/src/frontends/paddle/src/op/partial_concat.cpp
@@ -18,7 +18,7 @@ NamedOutputs partial_concat(const NodeContext& node) {
     int end_index;
     if (length < 0) {
         // Negative values for all elements after start_index on second dim.
-        end_index = x[0].get_shape()[1];
+        end_index = static_cast<int>(x[0].get_shape()[1]);
     } else {
         end_index = start_index + length;
     }

diff --git a/src/frontends/paddle/src/op/partial_sum.cpp b/src/frontends/paddle/src/op/partial_sum.cpp
@@ -11,15 +11,14 @@ namespace paddle {
 namespace op {
 NamedOutputs partial_sum(const NodeContext& node) {
     auto x = node.get_ng_inputs("X");
-    // const auto input_dtype = x.get_element_type();
     const auto start_index = node.get_attribute<int>("start_index");
     const auto length = node.get_attribute<int>("length");
     // PADDLE_OP_CHECK(node, x[0].get_shape.size() == 2, "partial op only supports 2-D Tensor");
 
     int end_index;
     if (length < 0) {
         // Negative values for all elements after start_index on second dim.
-        end_index = x[0].get_shape()[1];
+        end_index = static_cast<int>(x[0].get_shape()[1]);
     } else {
         end_index = start_index + length;
     }