AC: move tensor mapping to read network (#3401)

* AC: move tensor mapping to read network

* update work with batch

Author: eaidova

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/openvino_launcher.py

@@ -547,27 +547,29 @@ def _set_batch_size(self, batch_size):
             if layer_name in self.const_inputs:
                 input_shapes[layer_name] = parse_partial_shape(input_node.get_node().partial_shape)
             else:
-                layer_shape = parse_partial_shape(input_node.get_node().partial_shape)
-                layout = self.inputs[layer_name].layout
-                if '...' in str(layout):
-                    layout = self.get_layout_from_config(layer_name)
-                else:
-                    layout = str(layout).replace('[', '').replace(']', '').replace(',', '')
+                layer_shape = list(parse_partial_shape(input_node.get_node().partial_shape))
+                layout = self._process_layout(self.inputs[layer_name].layout, layer_name)
                 batch_pos = layout.find('N')
                 if batch_pos != -1:
                     layer_shape[batch_pos] = batch_size
                 input_shapes[layer_name] = layer_shape
         self._reshape_input(input_shapes, batch_size == -1)
         self._batch = batch_size
 
+    def _process_layout(self, ov_layout, layer_name):
+        if '...' in str(ov_layout) or ov_layout is None:
+            ov_layout = self.get_layout_from_config(layer_name)
+        else:
+            ov_layout = str(ov_layout).replace('[', '').replace(']', '').replace(',', '')
+        return ov_layout
+
     def _get_model_batch_size(self):
         input_nodes = self.network.inputs if self.network else self.exec_network.inputs
         input_info = input_nodes[0]
-        layout = input_info.get_node().layout
-        if '...' in str(layout) or layout is None:
-            layout = self.get_layout_from_config(input_info.get_node().friendly_name)
-        else:
-            layout = str(layout).replace('[', '').replace(']', '').replace(',', '')
+        layout = (
+            self._process_layout(input_info.get_node().layout, input_info.get_node().friendly_name)
+            or self.default_layout
+        )
         batch_pos = layout.find('N')
         if batch_pos != -1:
             return parse_partial_shape(input_info.partial_shape)[batch_pos]
@@ -702,6 +704,8 @@ def read_network(self, model, weights):
             network = self.ie_core.read_model(model=str(model), weights=str(weights))
         else:
             network = self.ie_core.read_model(model=str(model))
+        self.input_to_tensor_name = self.get_input_tensor_name_mapping(network)
+        self.input_to_index = {inp.get_node().friendly_name: idx for idx, inp in enumerate(network.inputs)}
         return network
 
     def inputs_info_for_meta(self, inputs=None):