Home
last modified time | relevance | path

Searched defs:FakeTensor (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_tensor.py279 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 Djit_compile_runtime_wrappers.py256 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__.py263 Optional[FakeTensor], List[Optional[FakeTensor]], Tuple[Optional[FakeTensor]]
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dgraph_builder.py51 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.py977 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 Dexport.ir_spec.rst355 FakeTensor section in References
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dfreezing.py69 example_inputs: List[torch._subclasses.FakeTensor],
/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/
H A D_tp_transform.py331 def create_output_spec(tensor: FakeTensor) -> DTensorSpec:
/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A Donnxruntime.py378 def _to_real_tensor(tensor: FakeTensor) -> torch.Tensor:
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A Dstage.py66 example: Union[torch.Tensor, FakeTensor], argument
/aosp_15_r20/external/executorch/exir/serde/
H A Dexport_serialize.py279 def _reduce_fake_tensor(fake_tensor: FakeTensor):
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dserialize.py271 def _reduce_fake_tensor(fake_tensor: FakeTensor):