diff --git a/src/core/src/op/fake_quantize.cpp b/src/core/src/op/fake_quantize.cpp index 0f192a6c261..ce59d271a1e 100644 --- a/src/core/src/op/fake_quantize.cpp +++ b/src/core/src/op/fake_quantize.cpp @@ -94,11 +94,11 @@ bool evaluate(const HostTensorPtr& arg0, arg3->get_data_ptr(), arg4->get_data_ptr(), out->get_data_ptr(), - parent->get_input_shape(0), - parent->get_input_shape(1), - parent->get_input_shape(2), - parent->get_input_shape(3), - parent->get_input_shape(4), + arg0->get_shape(), + arg1->get_shape(), + arg2->get_shape(), + arg3->get_shape(), + arg4->get_shape(), parent->get_levels(), parent->get_auto_broadcast()); return true;