Searched refs:dot_graph_shape (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/fx/passes/ |
| D | graph_drawer.py | 78 dot_graph_shape: Optional[str] = None,
82 self.dot_graph_shape = (
83 dot_graph_shape if dot_graph_shape is not None else "record"
86 _WEIGHT_TEMPLATE["shape"] = self.dot_graph_shape
153 "shape": self.dot_graph_shape,
439 dot_graph_shape: Optional[str] = None,
|
| /external/pytorch/torch/_inductor/ |
| D | debug.py | 95 gm, fname, clear_meta=False, dot_graph_shape=config.trace.dot_graph_shape
521 dot_graph_shape=config.trace.dot_graph_shape,
|
| D | config.py | 1197 dot_graph_shape = os.environ.get("INDUCTOR_DOT_GRAPH_SHAPE_SVG", None) variable in trace
|
| /external/pytorch/torch/_functorch/ |
| D | partitioners.py | 1910 dot_graph_shape: Optional[str] = None,
1925 dot_graph_shape=dot_graph_shape,
|