Searched refs:observe_tensor_cycles (Results 1 – 2 of 2) sorted by relevance
417 def observe_tensor_cycles(callback): function448 return observe_tensor_cycles(write_and_log)
79 from torch.utils.viz._cycles import observe_tensor_cycles3405 disarm = observe_tensor_cycles(observer)