Home
last modified time | relevance | path

Searched refs:parameter_proxy_cache (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/
H A D_symbolic_trace.py528 def getattr(self, attr: str, attr_val: Any, parameter_proxy_cache: Dict[str, Any]):
551 attr_val, collection_to_search, parameter_proxy_cache argument
555 if n not in parameter_proxy_cache:
569 parameter_proxy_cache[n] = val_proxy
570 return parameter_proxy_cache[n]
575 attr_val, self.root.named_parameters(), parameter_proxy_cache
582 attr_val, self.root.named_buffers(), parameter_proxy_cache
780 parameter_proxy_cache: Dict[
789 return self.getattr(attr, attr_val, parameter_proxy_cache)
/aosp_15_r20/external/executorch/exir/
H A Dtracer.py421 self, attr: str, attr_val: Value, parameter_proxy_cache: Dict[str, torch.Tensor]
426 if n not in parameter_proxy_cache:
428 parameter_proxy_cache[n] = PythonTensor(attr_val, proxy)
429 return parameter_proxy_cache[n]
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dmeta_tracer.py203 def getattr(self, attr, attr_val, parameter_proxy_cache): argument
207 return super().getattr(attr, attr_val, parameter_proxy_cache)
H A Dproxy_tensor.py1043 self, attr: str, attr_val: object, parameter_proxy_cache: Dict[str, Proxy]
1614 self, attr: str, attr_val: object, parameter_proxy_cache: Dict[str, Proxy]
1617 return super().getattr(attr, attr_val, parameter_proxy_cache)