def __init__(self, ch, env):
src = clip_head(inspect.getsource(ch.forward))
dprint(src)
self.ast = gast.ast_to_gast(ast.parse(src)).body[0]
self.call = User_Defined_Func_In_Link(ch, ch.forward).call
# 以下、 最初の外からのためのやつ
# code.InteractiveConsole({'v': self.ast}).interact()
self.forward_arglen = len(self.ast.args.args) - 1
# ここで、初期化したやつを上書きしてやる必要が出てくる
# あとでchainerで実行するために回復しないといけないので、
# restore_funcs に復元すべきものを追加している
self.inits = []
for s, v in ch.namedparams():
s = s[1:]
if s.find('/') != -1:
continue
t = helper.make_tensor_value_info('/' + s, TensorProto.FLOAT,
list(v.shape))
self.inits.append(t)
mv = getattr(ch, s)
setattr(ch, s, t)
env.restore_funcs.append(lambda: setattr(ch, s, mv))
# TODO(satos) Yieldをコンパイルできるとこれを消せる
mv = getattr(ch, 'children')
setattr(ch, 'children', Func(lambda _, __, ___: mv()))
env.restore_funcs.append(lambda: setattr(ch, 'children', mv))
def __init__(self, func):
self.func = func
src = clip_head(inspect.getsource(func))
dprint(src)
self.ast = gast.ast_to_gast(ast.parse(src)).body[0]
assert (isinstance(self.ast, gast.gast.FunctionDef))