Home
last modified time | relevance | path

Searched refs:shape_env_to_source_to_symbol_cache (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dbuilder.py2371 shape_env_to_source_to_symbol_cache = (
2372 prior_policy.shape_env_to_source_to_symbol_cache if prior_policy else None
2404 shape_env_to_source_to_symbol_cache=outer_context.shape_env_to_source_to_symbol_cache,
2416 shape_env_to_source_to_symbol_cache=shape_env_to_source_to_symbol_cache,
2434 shape_env_to_source_to_symbol_cache=shape_env_to_source_to_symbol_cache,
2668 shape_env_to_source_to_symbol_cache=shape_env_to_source_to_symbol_cache,
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dsymbolic_shapes.py1364shape_env_to_source_to_symbol_cache : Dict[int, Dict["TensorPropertySource", "sympy.Expr"]] = None variable in StatefulSymbolicContext
1370 if not self.shape_env_to_source_to_symbol_cache:
3440 and id(self) not in symbolic_context.shape_env_to_source_to_symbol_cache):
3441 symbolic_context.shape_env_to_source_to_symbol_cache[id(self)] = {}
3445 … and (source_name in symbolic_context.shape_env_to_source_to_symbol_cache[id(self)])):
3446 return symbolic_context.shape_env_to_source_to_symbol_cache[id(self)][source_name]
3453 symbolic_context.shape_env_to_source_to_symbol_cache[id(self)][source_name] = out
3472 symbolic_context.shape_env_to_source_to_symbol_cache[id(self)][source_name] = out
3583 symbolic_context.shape_env_to_source_to_symbol_cache[id(self)][source_name] = r