Python extract_port_from_string Examples

Example #1

File: graph.py Project: zoeysgithub/openvino

def get_name_and_port(tensor_name):
    node_name, in_port, out_port = extract_port_from_string(tensor_name)

    assert in_port is None or out_port is None

    if in_port is not None:
        return node_name, in_port
    elif out_port is not None:
        return node_name, out_port
    else:
        return node_name, 0

Example #2

File: extractor_test.py Project: groove-x/openvino

 def test_no_port(self):
     name, in_port, out_port = extract_port_from_string('node_name')
     self.assertEqual(name, 'node_name')
     self.assertEqual(in_port, None)
     self.assertEqual(out_port, None)