Searched refs:record_nn_module_stack (Results 1 – 1 of 1) sorted by relevance
83 def record_nn_module_stack(module_key: str, source, tx, mod: torch.nn.Module): function349 with record_nn_module_stack(495 with record_nn_module_stack(892 record_nn_module_stack(