update converter

This commit is contained in:
zwy 2020-04-30 23:53:57 +08:00
parent 9b0a4c1d8c
commit f9f0190590
1 changed files with 22 additions and 0 deletions

View File

@ -296,6 +296,28 @@ pjmap = {
}
}
def pjmap_append(pytorch_func_name, pytorch_args, jittor_func_module, jittor_func_name, jittor_args, extras=None, links=None, delete=None):
if links == None: links = {}
if extras == None: extras = {}
if delete == None: delete = []
assert isinstance(links, dict)
assert isinstance(extras, dict)
assert isinstance(delete, list)
pjmap[pytorch_func_name] = {
'pytorch': {
'args': pytorch_args,
},
'jittor': {
'module': jittor_func_module,
'name': jittor_func_name,
'args': jittor_args,
},
'links': links,
'extras': extras,
'delete': delete,
}
unsupport_ops = [
# ***************************************************************
# torch.nn