Searched defs:hook2 (Results 1 – 3 of 3) sorted by relevance
1029 auto hook2 = ([&hook2_count](Variable grad) { in TEST() local1055 auto hook2 = ([&hook2_count](Variable grad) { in TEST() local1094 auto hook2 = ([&hook2_count](Variable grad) { in TEST() local
62 def hook2(grads): function
1262 def hook2(tensor): function1433 def hook2(gs): function