Searched refs:logger_cls (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/ao/ns/fx/ |
| D | graph_passes.py | 45 logger_cls: Callable,
72 logger_obj = logger_cls(
94 logger_cls: Callable,
137 logger_cls,
157 logger_cls,
183 logger_cls,
243 logger_cls: Callable,
269 node_a, gm_a, logger_cls, node_type_to_io_type_map
272 node_c, gm_b, logger_cls, node_type_to_io_type_map
714 logger_cls: Callable,
[all …]
|
| D | utils.py | 41 logger_cls: Callable,
68 first_arg, gm, logger_cls, node_type_to_io_type_map
82 isinstance(mod, (logger_cls, ObserverBase, FakeQuantizeBase)) # type: ignore[arg-type]
93 first_arg, gm, logger_cls, node_type_to_io_type_map
120 prev_node, gm, logger_cls, node_type_to_io_type_map
135 prev_node, gm, logger_cls, node_type_to_io_type_map
152 first_arg, gm, logger_cls, node_type_to_io_type_map
|
| D | n_shadows_utils.py | 219 logger_cls: Callable,
229 logger_mod_orig = logger_cls(
|
| /external/pytorch/torch/ao/ns/ |
| D | _numeric_suite.py | 256 def __init__(self, q_module, float_module, logger_cls): argument
261 self.logger = logger_cls()
349 logger_cls: Callable,
383 prepare_model_with_stubs(float_mod, mod, module_swap_list, logger_cls)
390 reassign[name] = Shadow(mod, float_mod, logger_cls)
408 logger_cls=ShadowLogger, argument
442 prepare_model_with_stubs(float_model, q_model, module_swap_list, logger_cls)
483 logger_cls=OutputLogger, argument
501 qconfig_debug = torch.ao.quantization.QConfig(activation=logger_cls, weight=None)
520 logger_cls=OutputLogger, argument
[all …]
|
| D | _numeric_suite_fx.py | 448 logger_cls: Callable,
467 logger_cls,
478 logger_cls: Callable,
517 logger_cls,
524 logger_cls,
534 logger_cls: Callable,
578 logger_cls,
588 logger_cls: Callable,
595 is_logger = isinstance(mod, logger_cls) or ( # type: ignore[arg-type]
643 logger_cls: Callable,
[all …]
|