Searched refs:OutputLogger (Results 1 – 10 of 10) sorted by relevance
| /external/pytorch/test/quantization/fx/ |
| D | test_numeric_suite_fx.py | 76 OutputLogger,
878 'a', m, 'b', copy.deepcopy(mp), OutputLogger,
881 'a', mp, 'b', mq, OutputLogger,
909 m1, m2, OutputLogger, 'b')
946 OutputLogger, should_log_inputs=should_log_inputs)
948 'a', mp, 'b', mq, OutputLogger,
974 model, OutputLogger, 'b')
1087 ns.call_module(OutputLogger): 2,
1108 ns.call_module(OutputLogger): 2,
1148 mp_ns, mq_ns = add_loggers('a', mp, 'b', mq, OutputLogger)
[all …]
|
| /external/pytorch/torch/ao/quantization/pt2e/ |
| D | _numeric_debugger.py | 35 class OutputLogger(torch.nn.Module): class
84 OutputLogger(debug_handle, node.name, node.meta.get("nn_module_stack")),
189 if isinstance(module, OutputLogger) and len(module.stats) > 0:
|
| /external/pytorch/torch/quantization/ |
| D | _numeric_suite_fx.py | 24 OutputLogger,
|
| D | _numeric_suite.py | 23 OutputLogger,
|
| /external/pytorch/test/quantization/pt2e/ |
| D | test_numeric_debugger.py | 161 from torch.ao.quantization.pt2e._numeric_debugger import OutputLogger
163 loggers = [m for m in m_logger.modules() if isinstance(m, OutputLogger)]
|
| /external/pytorch/torch/ao/ns/ |
| D | _numeric_suite.py | 214 class OutputLogger(Logger): class
483 logger_cls=OutputLogger,
520 logger_cls=OutputLogger,
|
| D | _numeric_suite_fx.py | 141 class OutputLogger(nn.Module): class
249 class OutputComparisonLogger(OutputLogger):
1072 if isinstance(child, OutputLogger):
1086 if isinstance(child, OutputLogger):
1120 _extract_logger_info_one_model(model, results, OutputLogger)
|
| /external/pytorch/torch/ao/ns/fx/ |
| D | n_shadows_utils.py | 475 from torch.ao.ns._numeric_suite_fx import OutputComparisonLogger, OutputLogger
489 OutputLogger,
751 from torch.ao.ns._numeric_suite_fx import OutputComparisonLogger, OutputLogger
845 OutputLogger,
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | _equalize.py | 898 ns.OutputLogger,
906 model_a_ns, model_b_ns, ns.OutputLogger, "int8"
|
| /external/pytorch/test/quantization/eager/ |
| D | test_numeric_suite_eager.py | 17 OutputLogger,
518 logger = OutputLogger()
|