Searched refs:GraphTransformObserver (Results 1 – 10 of 10) sorted by relevance
8 from torch.fx.passes.graph_transform_observer import GraphTransformObserver37 with GraphTransformObserver(traced, "replace_neg_with_relu", log_url) as ob:43 current_pass_count = GraphTransformObserver.get_current_pass_count()
15 from torch.fx.passes.graph_transform_observer import GraphTransformObserver257 with GraphTransformObserver(301 with GraphTransformObserver(305 with GraphTransformObserver(309 with GraphTransformObserver(318 with GraphTransformObserver(322 with GraphTransformObserver(
6 from torch.fx.passes.graph_transform_observer import GraphTransformObserver30 with GraphTransformObserver(
14 from torch.fx.passes.graph_transform_observer import GraphTransformObserver442 with GraphTransformObserver(453 with GraphTransformObserver(466 with GraphTransformObserver(
25 from torch.fx.passes.graph_transform_observer import GraphTransformObserver585 with GraphTransformObserver(
21 from torch.fx.passes.graph_transform_observer import GraphTransformObserver89 with GraphTransformObserver(128 with GraphTransformObserver(
22 from torch.fx.passes.graph_transform_observer import GraphTransformObserver1312 with GraphTransformObserver(
15 class GraphTransformObserver: class23 GraphTransformObserver.__pass_count += 1
66 from torch.fx.passes.graph_transform_observer import GraphTransformObserver1614 with GraphTransformObserver(gm, msg, config.trace.log_url_for_graph_xform):
87 from torch.fx.passes.graph_transform_observer import GraphTransformObserver1702 with GraphTransformObserver(