Home
last modified time | relevance | path

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

/external/pytorch/test/dynamo/
Dtest_reorder_logs.py34 @torch._dynamo.config.patch(reorderable_logging_functions={print})
55 @torch._dynamo.config.patch(reorderable_logging_functions={warnings.warn})
77 @torch._dynamo.config.patch(reorderable_logging_functions={print})
113 with torch._dynamo.config.patch(reorderable_logging_functions={custom_log}):
121 @torch._dynamo.config.patch(reorderable_logging_functions={print})
/external/pytorch/torch/_dynamo/
Dconfig.py437 reorderable_logging_functions: Set[Callable[[Any], None]] = set() variable
/external/pytorch/torch/export/
D_trace.py103 reorderable_logging_functions: Set[Callable] = dataclasses.field( variable in ExportDynamoConfig
136 DEFAULT_EXPORT_DYNAMO_CONFIG.reorderable_logging_functions = {
/external/pytorch/torch/_dynamo/variables/
Dmisc.py1407 and obj in torch._dynamo.config.reorderable_logging_functions