Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dgraph_module.py48 class ObservedGraphModule(GraphModule): class
82 return ObservedGraphModule(
101 class ObservedStandaloneGraphModule(ObservedGraphModule):
/aosp_15_r20/external/pytorch/torch/quantization/fx/
H A Dgraph_module.py14 ObservedGraphModule,
/aosp_15_r20/external/pytorch/torch/quantization/
H A Dquantize_fx.py10 from torch.ao.quantization.fx.graph_module import ObservedGraphModule
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dquantize_fx.py13 from .fx.graph_module import ObservedGraphModule # noqa: F401