Fix wrong usage of convolution weight index

Signed-off-by: Shoujiang Ma <[email protected]>
openvinotoolkit · May 19, 2021 · 485e217 · 485e217
1 parent 04874aa
commit 485e217
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/src/auto_plugin/auto_plugin.cpp b/inference-engine/src/auto_plugin/auto_plugin.cpp
@@ -32,7 +32,7 @@ namespace {
                 std::dynamic_pointer_cast<ngraph::opset1::ConvolutionBackpropData>(node) ||
                 std::dynamic_pointer_cast<ngraph::op::ConvolutionIE>(node) ||
                 std::dynamic_pointer_cast<ngraph::op::DeconvolutionIE>(node)) {
-                auto layerType = node->input(0).get_element_type().get_type_name();
+                auto layerType = node->input(1).get_element_type().get_type_name();
                 if (layerType == "f32")
                     return METRIC_VALUE(FP32);
                 if (layerType == "f16")