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, {}
|