From ab4a11b3bde112d9b6cd47ff9aaa299bce8a2ed0 Mon Sep 17 00:00:00 2001 From: Mateusz Tabaka Date: Thu, 17 Feb 2022 06:51:32 +0100 Subject: [PATCH] Remove unnecessary AutoBroadcastSpec parameter in MatMulMultiplyFusion (#10005) --- .../common_optimizations/matmul_multiply_fusion.cpp | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/common/transformations/src/transformations/common_optimizations/matmul_multiply_fusion.cpp b/src/common/transformations/src/transformations/common_optimizations/matmul_multiply_fusion.cpp index d6da2f45f4d..0dc61b235a4 100644 --- a/src/common/transformations/src/transformations/common_optimizations/matmul_multiply_fusion.cpp +++ b/src/common/transformations/src/transformations/common_optimizations/matmul_multiply_fusion.cpp @@ -16,8 +16,7 @@ NGRAPH_RTTI_DEFINITION(pass::MatMulMultiplyFusion, "MatMulMultiplyFusion", 0); static std::shared_ptr fuse_const_to_weights(const std::shared_ptr& matmul, const Output& weights, - std::shared_ptr mul_const, - const op::AutoBroadcastSpec& autob) { + std::shared_ptr mul_const) { auto const_shape = mul_const->get_shape(); auto const_rank = static_cast(const_shape.size()); const auto& weights_shape = weights.get_partial_shape(); @@ -149,15 +148,13 @@ pass::MatMulMultiplyFusion::MatMulMultiplyFusion() { matcher_pass_callback callback = [=](pattern::Matcher& m) { const auto& pattern_map = m.get_pattern_value_map(); const auto& weights = pattern_map.at(weights_pattern); - auto mul = std::dynamic_pointer_cast(pattern_map.at(mul_pattern).get_node_shared_ptr()); - if (!mul) - return false; + auto mul = pattern_map.at(mul_pattern).get_node_shared_ptr(); auto mul_const = std::dynamic_pointer_cast(pattern_map.at(mul_const_pattern).get_node_shared_ptr()); if (!mul_const) return false; auto matmul = pattern_map.at(matmul_pattern).get_node_shared_ptr(); - auto new_weights = fuse_const_to_weights(matmul, weights, mul_const, mul->get_autob()); + auto new_weights = fuse_const_to_weights(matmul, weights, mul_const); if (!new_weights) return false;