def generate_value_with_same_type(value : 'values.Value'):
if isinstance(value, values.TensorValue):
ret = values.TensorValue()
ret.shape = value.shape
return ret
if isinstance(value, values.NumberValue):
ret = values.NumberValue(None)
return ret
if isinstance(value, values.StrValue):
ret = values.StrValue(None)
return ret
if isinstance(value, values.BoolValue):
ret = values.BoolValue(None)
return ret
if isinstance(value, values.ListValue):
ret = values.ListValue(None)
return ret
if isinstance(value, values.NoneValue):
ret = values.NoneValue()
return ret
if isinstance(value, values.Value):
ret = values.Value()
return ret
return None
def generate_value_with_same_type(value: 'values.Value'):
assert (isinstance(value, values.Value))
ret = None
if isinstance(value, values.TensorValue):
ret = values.TensorValue()
ret.shape = value.shape
if isinstance(value, values.NumberValue):
ret = values.NumberValue(None)
if isinstance(value, values.StrValue):
ret = values.StrValue(None)
if isinstance(value, values.BoolValue):
ret = values.BoolValue(None)
if isinstance(value, values.ListValue):
ret = values.ListValue(None)
if isinstance(value, values.NoneValue):
ret = values.NoneValue()
if ret is None and isinstance(value, values.Value):
ret = values.Value()
if ret is not None:
ret.name = value.name + '_st'
return ret
def generate_copied_value(value : 'values.Value'):
if isinstance(value, values.NumberValue):
copied = values.NumberValue(value.internal_value)
return copied
if isinstance(value, values.TensorValue):
copied = values.TensorValue()
copied.shape = value.shape
return copied
if isinstance(value, values.NoneValue):
copied = values.NoneValue()
return copied
if isinstance(value, values.BoolValue):
copied = values.BoolValue(value.internal_value)
return copied
if isinstance(value, values.StrValue):
copied = values.StrValue(value.internal_value)
return copied
if isinstance(value, values.RangeValue):
copied = values.RangeValue()
return copied
if isinstance(value, values.TupleValue):
copied = values.TupleValue(value.values)
return copied
if config.show_warnings:
print('Unknown type {} is copied'.format(value))
return values.Value()
def __init__(self):
super().__init__()
self.name = 'full'
self.args.add_arg('shape', values.NoneValue())
self.args.add_arg('fill_value', values.NoneValue())
self.args.add_arg('dtype', values.NoneValue())
self.args.add_arg('order', values.StrValue('C'))
def __init__(self):
super().__init__()
self.name = 'array'
fa = functions.FunctionArg()
fa.name = 'object'
fa.obj = values.Object(values.NoneValue())
self.funcArgs.append(fa)
fa = functions.FunctionArg()
fa.name = 'dtype'
fa.obj = values.Object(values.NoneValue())
self.funcArgs.append(fa)
fa = functions.FunctionArg()
fa.name = 'copy'
fa.obj = values.Object(values.BoolValue(True))
self.funcArgs.append(fa)
fa = functions.FunctionArg()
fa.name = 'order'
fa.obj = values.Object(values.StrValue('K'))
self.funcArgs.append(fa)
fa = functions.FunctionArg()
fa.name = 'subok'
fa.obj = values.Object(values.BoolValue(False))
self.funcArgs.append(fa)
fa = functions.FunctionArg()
fa.name = 'ndmin'
fa.obj = values.Object(values.NumberValue(0))
self.funcArgs.append(fa)
def generate_value_with_same_type(value: 'values.Value'):
assert (isinstance(value, values.Value))
ret = None
if isinstance(value, values.TensorValue):
ret = values.TensorValue()
ret.shape = value.shape
if isinstance(value, values.NumberValue):
ret = values.NumberValue(None)
if value.internal_value is None:
ret.dtype = value.dtype
elif isinstance(value.internal_value, int):
ret.dtype = np.array(value.internal_value).dtype
elif isinstance(value.internal_value, float):
ret.dtype = np.array(value.internal_value).dtype
if isinstance(value, values.StrValue):
ret = values.StrValue(None)
if isinstance(value, values.BoolValue):
ret = values.BoolValue(None)
if isinstance(value, values.ListValue):
ret = values.ListValue(None)
if isinstance(value, values.NoneValue):
ret = values.NoneValue()
if ret is None and isinstance(value, values.Value):
ret = values.Value()
if ret is not None:
ret.name = value.name + '_st'
return ret
def veval_ast_str(astc: 'AstContext', local_field: 'values.Field',
graph: 'Graph'):
'''
Ex. "str"
'''
assert (isinstance(astc.nast, gast.gast.Str))
lineprop = utils.LineProperty(astc.lineno)
value = values.StrValue(astc.nast.s)
return values.Object(value)
def __init__(self):
super().__init__()
self.name = 'array'
self.args.add_arg('object', values.NoneValue())
self.args.add_arg('dtype', values.NoneValue())
self.args.add_arg('copy', values.BoolValue(True))
self.args.add_arg('order', values.StrValue('K'))
self.args.add_arg('subok', values.BoolValue(False))
self.args.add_arg('ndmin', values.NumberValue(0))
def veval_ast_str(astc: 'AstContext', local_field: 'values.Field',
graph: 'Graph'):
'''
Ex. "str"
'''
assert (isinstance(astc.nast, gast.gast.Str))
lineprop = utils.LineProperty(astc.lineno)
value = values.StrValue(astc.nast.s)
ret = values.ValueRef(value)
name = values.create_ref_value_name_with_constant(ret)
ret.name = name
ret.get_value().name = name
return ret
def generate_copied_value(value: 'values.Value'):
assert (isinstance(value, values.Value))
if isinstance(value, values.NumberValue):
copied = values.NumberValue(value.internal_value)
copied.dtype = value.dtype
return copied
if isinstance(value, values.TensorValue):
copied = values.TensorValue()
copied.value = value.value
copied.shape = value.shape
copied.dtype = value.dtype
return copied
if isinstance(value, values.ListValue):
copied = values.ListValue()
copied.is_any = value.is_any
if value.internal_value is not None:
copied.internal_value = value.internal_value.copy()
return copied
if isinstance(value, values.NoneValue):
copied = values.NoneValue()
return copied
if isinstance(value, values.BoolValue):
copied = values.BoolValue(value.internal_value)
return copied
if isinstance(value, values.StrValue):
copied = values.StrValue(value.internal_value)
return copied
if isinstance(value, values.RangeValue):
copied = values.RangeValue()
return copied
if isinstance(value, values.TupleValue):
if value.internal_value is not None:
copied = values.TupleValue(value.internal_value.copy())
else:
copied = values.TupleValue(value.internal_value)
return copied
if config.show_warnings:
print('Unknown type {} is copied'.format(value))
return values.Value()
def generate_value_with_same_type(value: 'values.Value',
is_dummy_value=False,
suffix_type=SuffixType.Unknown):
assert (isinstance(value, values.Value))
ret = None
if isinstance(value, values.TensorValue):
ret = values.TensorValue()
ret.shape = value.shape
ret.dtype = value.dtype
elif isinstance(value, values.NumberValue):
dtype = None
if value.internal_value is None:
dtype = value.dtype
elif isinstance(value.internal_value, int):
dtype = np.array(value.internal_value).dtype
elif isinstance(value.internal_value, float):
dtype = np.array(value.internal_value).dtype
ret = values.NumberValue(None)
ret.dtype = dtype
elif isinstance(value, values.StrValue):
ret = values.StrValue(None)
elif isinstance(value, values.BoolValue):
ret = values.BoolValue(None)
elif isinstance(value, values.ListValue):
ret = values.ListValue(None)
elif isinstance(value, values.NoneValue):
ret = values.NoneValue()
elif isinstance(value, values.TupleValue):
ret = values.TupleValue()
elif isinstance(value, values.RangeValue):
ret = values.RangeValue()
elif isinstance(value, values.UnknownValue):
ret = values.UnknownValue()
elif ret is None and isinstance(value, values.Value):
ret = values.Value()
else:
assert (False)
assert (ret is not None)
ret.is_dummy_value = is_dummy_value
if suffix_type == SuffixType.Unknown:
ret.name = value.name + '_st'
elif suffix_type == SuffixType.Unused:
ret.name = value.name + '_unused'
elif suffix_type == SuffixType.Dummy:
ret.name = value.name + '_dummy'
elif suffix_type == SuffixType.Input:
ret.name = value.name + '_in'
else:
assert (False)
return ret
def generate_value_with_same_type(value: 'values.Value',
has_default=False,
suffix_type=SuffixType.Unknown):
assert (isinstance(value, values.Value))
ret = None
if isinstance(value, values.TensorValue):
ret = values.TensorValue()
ret.shape = value.shape
ret.dtype = value.dtype
if isinstance(value, values.NumberValue):
dtype = None
if value.internal_value is None:
dtype = value.dtype
elif isinstance(value.internal_value, int):
dtype = np.array(value.internal_value).dtype
elif isinstance(value.internal_value, float):
dtype = np.array(value.internal_value).dtype
if has_default:
if dtype == np.array(0).dtype:
ret = values.NumberValue(0)
elif dtype == np.array(0.0).dtype:
ret = values.NumberValue(0.0)
else:
ret = values.NumberValue(None)
else:
ret = values.NumberValue(None)
ret.dtype = dtype
if isinstance(value, values.StrValue):
if has_default:
ret = values.StrValue('')
else:
ret = values.StrValue(None)
if isinstance(value, values.BoolValue):
if has_default:
ret = values.BoolValue(False)
else:
ret = values.BoolValue(None)
if isinstance(value, values.ListValue):
ret = values.ListValue(None)
if isinstance(value, values.NoneValue):
ret = values.NoneValue()
if isinstance(value, values.TupleValue):
ret = values.TupleValue()
if isinstance(value, values.UnknownValue):
ret = values.UnknownValue()
if has_default:
ret.internal_value = 0
if ret is None and isinstance(value, values.Value):
ret = values.Value()
if ret is not None:
if suffix_type == SuffixType.Unknown:
ret.name = value.name + '_st'
if suffix_type == SuffixType.Unused:
ret.name = value.name + '_unused'
return ret
