Home
last modified time | relevance | path

Searched refs:record_nn_module_stack (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dnn_module.py83 def record_nn_module_stack(module_key: str, source, tx, mod: torch.nn.Module): function
349 with record_nn_module_stack(
495 with record_nn_module_stack(
892 record_nn_module_stack(