Searched refs:complex_state_touching_hook (Results 1 – 1 of 1) sorted by relevance
444 def complex_state_touching_hook(grad, *, obj): function450 hook1 = functools.partial(complex_state_touching_hook, obj=obj)451 hook2 = functools.partial(complex_state_touching_hook, obj=obj)493 def complex_state_touching_hook(grad, *, obj): function499 hook1 = functools.partial(complex_state_touching_hook, obj=obj)500 hook2 = functools.partial(complex_state_touching_hook, obj=obj)