Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A 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})
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dconfig.py437 reorderable_logging_functions: Set[Callable[[Any], None]] = set() variable
/aosp_15_r20/external/pytorch/torch/export/
H A D_trace.py103 reorderable_logging_functions: Set[Callable] = dataclasses.field( variable in ExportDynamoConfig
136 DEFAULT_EXPORT_DYNAMO_CONFIG.reorderable_logging_functions = {
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dmisc.py1407 and obj in torch._dynamo.config.reorderable_logging_functions