def extend(op: Node):
def normalize_port_map(port_map: dict):
for port in port_map:
for elem in ['axis', 'stride', 'part_size', 'start', 'end']:
if port.get(elem) is None:
port[elem] = None
assert op.has('body'), 'Something wrong with TensorIterator layer {}, please check!'.format(op.name)
# Now op.body is an IREngine, we need to replace it with IREngine.graph
op.body.graph.graph['cmd_params'] = op.graph.graph['cmd_params']
op.body.graph.graph['ir_version'] = op.graph.graph['ir_version']
op.body.graph.name = op.name + '/body'
for node in op.body.graph.get_op_nodes():
node['internal_layer_id'] = int(node.id)
op.body = copy_graph_with_ops(op.body.graph)
normalize_port_map(op.input_port_map)
normalize_port_map(op.output_port_map)
for edge in op.back_edges:
edge['from_layer'] = edge['from-layer']
edge['to_layer'] = edge['to-layer']
del(edge['from-layer'])
del(edge['to-layer'])
op['infer'] = Extender.const_shape_infer
def extend(op: Node):
for attr in StridedSlice.get_mask_names():
# We can not use op.has_and_set(attr) here as a condition, because it will return False if begin/end is
# 1D tensor and begin_mask/end_mask is equal to 0
if op.has(attr) and op[attr] != '':
Extender.attr_to_list(op, attr)
else:
assert attr not in ['begin_mask', 'end_mask'],\
'{} is not defined for the node {}'.format(attr, op.soft_get('name', op.id))
op[attr] = int64_array([0])
op.begin_mask = int64_array([1 - i for i in op.begin_mask])
op.end_mask = int64_array([1 - i for i in op.end_mask])
def extend(op: Node):
assert op.has(
'then_graph'
), 'There is no "then_body" attribute in the If op {}.'.format(op.name)
assert op.has(
'else_graph'
), 'There is no "else_body" attribute in the If op {}.'.format(op.name)
# Now op.body is an IREngine, we need to replace it with IREngine.graph
op.then_graph.graph.graph['cmd_params'] = op.graph.graph['cmd_params']
op.then_graph.graph.graph['ir_version'] = op.graph.graph['ir_version']
op.then_graph.graph.name = op.name + '/then_body'
op.else_graph.graph.graph['cmd_params'] = op.graph.graph['cmd_params']
op.else_graph.graph.graph['ir_version'] = op.graph.graph['ir_version']
op.else_graph.graph.name = op.name + '/else_body'
op.then_graph = copy_graph_with_ops(op.then_graph.graph)
op.else_graph = copy_graph_with_ops(op.else_graph.graph)
IfExtender.set_input_output_id(op.then_graph, op.then_input_port_map,
op.then_output_port_map)
IfExtender.set_input_output_id(op.else_graph, op.else_input_port_map,
op.else_output_port_map)
def extend(op: Node):
def normalize_port_map(port_map: dict):
for port in port_map:
for elem in [
'axis', 'stride', 'part_size', 'start', 'end',
'purpose'
]:
if port.get(elem) is None:
port[elem] = None
assert op.has(
'body'), 'There is no "body" attribute in the Loop op {}.'.format(
op.name)
# Now op.body is an IREngine, we need to replace it with IREngine.graph
op.body.graph.graph['cmd_params'] = op.graph.graph['cmd_params']
op.body.graph.graph['ir_version'] = op.graph.graph['ir_version']
op.body.graph.name = op.name + '/body'
for node in op.body.graph.get_op_nodes():
node['internal_layer_id'] = int(node.id)
op.body = copy_graph_with_ops(op.body.graph)
normalize_port_map(op.input_port_map)
normalize_port_map(op.output_port_map)
# the 'external_port_id' uses end-to-end numbering of ports, but at this moment it is separate for input and
# output ports so we need to decrease the output por_id with a number of input ports
for record in op.output_port_map:
if record['external_port_id'] != -1:
record['external_port_id'] -= len(op.in_ports())
for edge in op.back_edges:
edge['from_layer'] = edge['from-layer']
edge['to_layer'] = edge['to-layer']
edge['to_port'] = 0
edge['from_port'] = 0
del (edge['from-layer'])
del (edge['to-layer'])