diff --git a/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.cpp b/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.cpp index 6dc1623547d..c5e52a89edb 100644 --- a/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.cpp +++ b/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.cpp @@ -63,9 +63,10 @@ Result NoBroadCastEltwiseShapeInfer::infer( if (input_shape.size() != output_shape.size()) { OPENVINO_THROW("Eltwise shape infer input and output shapes rank mismatch"); } - for (size_t j = 0; j < input_shapes.size(); ++j) { + for (size_t j = 0; j < input_shape.size(); ++j) { if (input_shape[j] != output_shape[j]) { - OPENVINO_THROW("Eltwise shape infer input shapes dim index: ", j, " mismatch"); + OPENVINO_THROW("Eltwise shape infer input shapes dim index: ", j, " mismatch", " first input:", ov::intel_cpu::vec2str(input_shape) + , " second input:", ov::intel_cpu::vec2str(output_shape)); } } return { { std::move(output_shape) }, ShapeInferStatus::success }; diff --git a/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.hpp b/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.hpp index f8bcc82c29e..39fde502904 100644 --- a/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.hpp +++ b/src/plugins/intel_cpu/src/shape_inference/custom/eltwise.hpp @@ -4,6 +4,7 @@ #include #include "shape_inference/shape_inference_cpu.hpp" +#include "transformations/cpu_opset/common/op/power_static.hpp" #pragma once namespace ov { @@ -41,7 +42,8 @@ public: EltwiseShapeInferFactory(std::shared_ptr op) : m_op(op) {} ShapeInferPtr makeShapeInfer() const override { const auto& autob = m_op->get_autob(); - if (autob.m_type == ov::op::AutoBroadcastType::NONE) { + if (autob.m_type == ov::op::AutoBroadcastType::NONE + && (!ov::is_type(m_op))) { return std::make_shared(); } else { return std::make_shared();