Searched refs:symbol_to_source (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | symbolic_shapes.py | 1559 symbol_to_source, argument
1564 self.symbol_to_source = symbol_to_source
1583 for symbol, sources in self.symbol_to_source.items()
1586 assert self.symbol_to_source.get(expr), (
1591 return self.source_ref(self.symbol_to_source[expr][0])
1607 def __init__(self, symbol_to_source, source_name_to_debug_name): argument
1609 self.symbol_to_source = symbol_to_source
1614 assert self.symbol_to_source.get(expr), (
1617 return self.symbol_to_source[expr][0].name()
1631 symbol_to_source, argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dynamic_shapes.py | 1610 symbol_to_source = defaultdict(list)
1611 symbol_to_source[s0].append(
1616 dcp = DynamicDimConstraintPrinter(symbol_to_source, {})
1680 symbol_to_source = {
1689 symbol_to_source, var_to_val, marked_dynamic, {}
|