| /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):
|