def extend(op: Node):
assert op.has_valid(
'element_type'
), 'Parameter node {} has missed element_type attr!'.format(op.name)
op['data_type'] = destination_type_to_np_data_type(op.element_type)
if op.shape == '':
op.shape = int64_array([])
else:
Extender.attr_to_list(op, 'shape')
if -1 in op.shape:
op.shape = shape_array([
d if d != -1 else dynamic_dimension_value for d in op.shape
])
def extend(op: Node):
assert op.has_valid(
'element_type'
), 'Parameter node {} has missed element_type attr!'.format(op.name)
op['data_type'] = destination_type_to_np_data_type(op.element_type)
if op.shape == '':
op.shape = int64_array([])
else:
Extender.attr_to_list(op, 'shape')
for i, dim in enumerate(op.shape):
if dim == -1 or (isinstance(dim, str) and ".." in dim):
op.shape[i] = -1
op.shape = shape_array(
[d if d != -1 else dynamic_dimension_value for d in op.shape])
def extend(op: Node):
assert op.has_valid(
'element_type'
), 'Parameter node {} has missed element_type attr!'.format(op.name)
op['data_type'] = destination_type_to_np_data_type(op.element_type)
if op.shape == '':
op.shape = int64_array([])
else:
Extender.attr_to_list(op, 'shape')
shape = op.shape.copy()
has_shapes_with_boundaries = False
for i, dim in enumerate(op.shape):
if dim == -1 or (isinstance(dim, str) and ".." in dim):
shape[i] = -1
if ".." in dim:
has_shapes_with_boundaries = True
shape = shape_array([
d if d not in [-1, '?'] else dynamic_dimension_value
for d in shape
])
if has_shapes_with_boundaries:
shape_list = []
for i, dim in enumerate(op.shape):
if not isinstance(dim, str):
shape_list.append(dim)
else:
shape_list.append(parse_dimension(dim))
# This value is used only for serialization of partial shapes with boundaries
# for Parameter node.
# 'user_shape' is not used in shape inference, as propagation of partial shapes with boundaries
# is not implemented in MO.
op['user_shape'] = tuple(shape_list)
# If 'user_shape' is not set, 'shape' attribute is used for serialization.
# 'shape' is also used for shape inference.
op.shape = shape