/aosp_15_r20/external/pytorch/torch/_subclasses/ |
H A D | fake_tensor.py | 279 def add_constant_storage_mapping(self, fake_tensor: FakeTensor) -> None: 314 def set_tensor_memo(self, t: Tensor, v: FakeTensor) -> None: 551 def _memo(self, obj: FakeTensor) -> str: 554 def _memo_vc(self, obj: FakeTensor) -> str: 561 def _memo_epoch(self, obj: FakeTensor) -> str: 565 self, obj: FakeTensor, objtype: Optional[Type[FakeTensor]] = None 581 def __set__(self, obj: FakeTensor, value: Optional[torch.SymInt]) -> None: 593 class FakeTensor(Tensor): class 1481 output: Optional[FakeTensor], 1634 output: Optional[FakeTensor], [all …]
|
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
H A D | jit_compile_runtime_wrappers.py | 256 fw_ins: List[Optional[FakeTensor]], 257 user_fw_outs: List[Optional[FakeTensor]], 258 bw_outs: List[Optional[FakeTensor]], 259 saved_tensors: List[FakeTensor],
|
/aosp_15_r20/external/executorch/exir/passes/ |
H A D | __init__.py | 263 Optional[FakeTensor], List[Optional[FakeTensor]], Tuple[Optional[FakeTensor]]
|
/aosp_15_r20/external/executorch/backends/cadence/aot/ |
H A D | graph_builder.py | 51 self, target: str, fake_tensor: Union[FakeTensor, torch.Tensor] 82 placeholders: Sequence[Union[torch.Tensor, FakeTensor]], argument
|
/aosp_15_r20/external/executorch/exir/emit/ |
H A D | _emitter.py | 977 val: Tuple[Union[torch.SymInt, torch.BoolType, torch.FloatType, FakeTensor]], argument 993 val: Tuple[Union[torch.SymInt, torch.BoolType, torch.FloatType, FakeTensor]], argument
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | export.ir_spec.rst | 355 FakeTensor section in References
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | freezing.py | 69 example_inputs: List[torch._subclasses.FakeTensor],
|
/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/ |
H A D | _tp_transform.py | 331 def create_output_spec(tensor: FakeTensor) -> DTensorSpec:
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
H A D | onnxruntime.py | 378 def _to_real_tensor(tensor: FakeTensor) -> torch.Tensor:
|
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/ |
H A D | stage.py | 66 example: Union[torch.Tensor, FakeTensor], argument
|
/aosp_15_r20/external/executorch/exir/serde/ |
H A D | export_serialize.py | 279 def _reduce_fake_tensor(fake_tensor: FakeTensor):
|
/aosp_15_r20/external/pytorch/torch/_export/serde/ |
H A D | serialize.py | 271 def _reduce_fake_tensor(fake_tensor: FakeTensor):
|