Fix numpy constant type in MO

openvinotoolkit · Feb 16, 2021 · f5289bb · f5289bb
1 parent 22169a0
commit f5289bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model-optimizer/extensions/front/tf/TFSliceToSlice.py b/model-optimizer/extensions/front/tf/TFSliceToSlice.py
@@ -44,7 +44,7 @@ def replace_sub_graph(self, graph: Graph, match: dict):
 
         eq_node = Equal(graph, {'name': slice_name + '/equal'}).create_node()
         minus_one_node = Const(graph, {'name': slice_name + '/minus_one', 'value': np.array(-1)}).create_node()
-        int32_max_node = Const(graph, {'name': slice_name + '/int32_max', 'value': np.iinfo(np.int32).max}).create_node()
+        int32_max_node = Const(graph, {'name': slice_name + '/int32_max', 'value': np.int64(np.iinfo(np.int32).max)}).create_node()
         select_node = Select(graph, {'name': slice_name + '/select'}).create_node()
 
         # node to convert sizes to ends