From 8e9eaaee9137b3ab867ef0e7d2ff663a2879189c Mon Sep 17 00:00:00 2001 From: Mateusz Bencer Date: Fri, 10 Jun 2022 16:57:03 +0200 Subject: [PATCH] [FakeQunatize] Use shape from HostTensor instead of FakeQuantize* (#11843) --- src/core/src/op/fake_quantize.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) 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;