zhousiyi a5b4e7bbf8 lift fv before grad except weight, then convert ... switch(cond, partial(g1, xs), partial(g2, ys))(Zs) to switch(cond, g1, g2)(Xs, Ys, Zs) switch_layer(index, make_tuple(partial(g1, xs), partial(g2, ys)))(Zs) to switch_layer(index, make_tuple(g1, g2))(Xs, Ys, Zs) put Zs at last when unifyparameter as it may have u-monad or io-monad use joined args other than broadened one as some extra parameter which is not a parameter of while_header can be add to while_body inline fprop_switch forcely reorder the parameter if one of the parameter is Monad when incorporate call incorporate switch tuple_getitem if item 0 of tuple is EnvInstance or item 1 of tuple is bprop function addn with shape() and shape(1) remove context_ from FuncGraphEvaluator to make it re-entry able to resolve evaluator stuck issue because of re-entry of the same FuncGraphEvaluator		2021-07-03 00:03:37 +08:00
..
__init__.py	pylint waring clean	2020-05-13 11:30:27 +08:00
test_auto_grad.py	lift fv before grad except weight, then convert	2021-07-03 00:03:37 +08:00
test_debug_location.py	support name or attribute ast.Expr	2021-03-11 21:40:33 +08:00
test_optimizer_with_loss_scale.py	Change TensorAdd to Add, from r1.1 to master	2021-02-01 17:53:52 +08:00
test_optimizer_with_parameter_groups.py	init parameter data by defaultOnly keep no data as MetaTensor in auto parallel mode	2020-08-06 19:52:31 +08:00
test_python_pass.py	Fix compile cache for resnet50	2021-06-24 14:36:24 +08:00
test_recompute.py	Fix calling recompute api after compiling	2021-02-08 16:19:49 +08:00
test_while_ScatterNdUpdate.py	add notice	2020-06-29 17:39:25 +08:00