Searched defs:FakeTensorMode (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_tensor.py | 326 fake_mode: FakeTensorMode,
462 self, fake_mode: FakeTensorMode, t: Tensor, device: torch.device
493 fake_mode: FakeTensorMode,
663 fake_mode: FakeTensorMode,
743 def from_tensor(t: Tensor, fake_mode: FakeTensorMode) -> FakeTensor:
938 mode: FakeTensorMode,
1069 class FakeTensorMode(TorchDispatchMode): class
2253 fake_mode: FakeTensorMode,
2319 def __init__(self, fake_mode: FakeTensorMode) -> None:
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | non_strict_utils.py | 84 mode: FakeTensorMode,
244 fake_mode: FakeTensorMode,
309 fake_mode: FakeTensorMode,
421 fake_mode: torch._subclasses.fake_tensor.FakeTensorMode,
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | aot_autograd.py | 437 fake_mode: FakeTensorMode,
518 fake_mode: FakeTensorMode,
531 fake_mode: FakeTensorMode,
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | fake_tensor_prop.py | 27 def __init__(self, module: torch.fx.GraphModule, mode: Optional[FakeTensorMode] = None):
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | out_dtype.py | 151 mode: FakeTensorMode,
|
| H A D | flex_attention.py | 457 mode: FakeTensorMode,
1029 mode: FakeTensorMode,
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | _trace.py | 774 fake_mode: FakeTensorMode,
1128 def _get_inline_constraints(fake_mode: FakeTensorMode):
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | eval_frame.py | 882 fake_mode: Optional[fake_tensor.FakeTensorMode] = None,
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 1833 prev_fake_tensor_mode: Optional[FakeTensorMode],
|