diff --git a/src/frontends/pytorch/src/op/index.cpp b/src/frontends/pytorch/src/op/index.cpp index c1c4c1f11dc71f..e985d2ee7a6ebe 100644 --- a/src/frontends/pytorch/src/op/index.cpp +++ b/src/frontends/pytorch/src/op/index.cpp @@ -54,7 +54,7 @@ OutputVector translate_index(const NodeContext& context) { } } if (index_ov_type == element::boolean || index_ov_type == element::u8) { - auto nonzero = context.mark_node(std::make_shared(indices, element::i32)); + auto nonzero = context.mark_node(std::make_shared(indices)); auto input_order = context.mark_node(v0::Constant::create(element::i32, Shape{2}, {1, 0})); auto masked_id = context.mark_node(std::make_shared(nonzero, input_order)); auto gather = context.mark_node(std::make_shared(x, masked_id)); diff --git a/src/frontends/pytorch/src/op/masked_scatter.cpp b/src/frontends/pytorch/src/op/masked_scatter.cpp index df3d93358c9e92..aa83cf1a3d1d8f 100644 --- a/src/frontends/pytorch/src/op/masked_scatter.cpp +++ b/src/frontends/pytorch/src/op/masked_scatter.cpp @@ -34,7 +34,7 @@ OutputVector translate_masked_scatter(const NodeContext& context) { auto x_shape = context.mark_node(std::make_shared(x, element::i32)); auto expanded_mask = context.mark_node(std::make_shared(mask, x_shape, BroadcastType::BIDIRECTIONAL)); - auto index = context.mark_node(std::make_shared(expanded_mask, element::i32)); + auto index = context.mark_node(std::make_shared(expanded_mask)); auto input_order = context.mark_node(v0::Constant::create(element::i32, Shape{2}, {1, 0})); index = context.mark_node(std::make_shared(index, input_order)); // source can be arbitary shape, select only relevant data diff --git a/src/frontends/pytorch/src/transforms/aten_index_put_replacer.cpp b/src/frontends/pytorch/src/transforms/aten_index_put_replacer.cpp index 015394d195a7a0..f962080a1d0e56 100644 --- a/src/frontends/pytorch/src/transforms/aten_index_put_replacer.cpp +++ b/src/frontends/pytorch/src/transforms/aten_index_put_replacer.cpp @@ -142,7 +142,7 @@ AtenIndexPutReplacer::AtenIndexPutReplacer() { auto input_shape = rg.make(input, element::i32); auto input_rank = rg.make(input_shape, element::i32); auto one_const = v0::Constant::create(element::i32, Shape{1}, {1}); - auto nonzero = rg.make(index, element::i32); + auto nonzero = rg.make(index); auto input_order = v0::Constant::create(element::i32, Shape{2}, {1, 0}); index = rg.make(nonzero, input_order); auto result = rg.make(input, index, values); diff --git a/src/frontends/pytorch/src/transforms/aten_index_replacer.cpp b/src/frontends/pytorch/src/transforms/aten_index_replacer.cpp index f22c7bacf8baf6..537cd36f658db3 100644 --- a/src/frontends/pytorch/src/transforms/aten_index_replacer.cpp +++ b/src/frontends/pytorch/src/transforms/aten_index_replacer.cpp @@ -76,7 +76,7 @@ AtenIndexToSelect::AtenIndexToSelect() { } auto index_dtype = indicies->get_output_element_type(0); if (index_dtype == element::boolean || index_dtype == element::u8) { - auto nonzero = rg.make(indicies, element::i32); + auto nonzero = rg.make(indicies); auto input_order = v0::Constant::create(element::i32, Shape{2}, {1, 0}); auto masked_id = rg.make(nonzero, input_order); auto gather = rg.make(input_node, masked_id); diff --git a/src/frontends/pytorch/src/utils.cpp b/src/frontends/pytorch/src/utils.cpp index 70ba4171770fbd..30bb4b496d6e8c 100644 --- a/src/frontends/pytorch/src/utils.cpp +++ b/src/frontends/pytorch/src/utils.cpp @@ -728,7 +728,7 @@ bool index_tensor_on_list(ov::pass::NodeRegistry& rg, auto id_dtype = indices[i].get_element_type(); if (id_dtype == element::boolean || id_dtype == element::u8) { auto idx = rg.make(indices[i], element::u8); - auto nonzero = rg.make(idx, element::i32); + auto nonzero = rg.make(idx); auto input_order = rg.make(element::i32, Shape{2}, std::vector{1, 0}); auto masked_id = rg.make(nonzero, input_order); masked_indicies.push_back(masked_id);