Searched refs:tensor_hook (Results 1 – 1 of 1) sorted by relevance
2653 def tensor_hook(_): function2667 loss.register_hook(tensor_hook)