| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | functional_tensor.py | 46 class FunctionalTensor(torch.Tensor): class
138 FunctionalTensor._extra_dispatch_keys & torch._C._dispatch_keys(elem)
186 if t not in [torch.Tensor, torch._subclasses.FakeTensor, FunctionalTensor]
200 if func in FunctionalTensor.metadata_fns:
209 assert len(args) == 2 and isinstance(args[0], FunctionalTensor)
211 assert len(args) == 1 and isinstance(args[0], FunctionalTensor)
249 out = FunctionalTensor(x_functional, functional_mode)
343 torch.storage.UntypedStorage, Optional[FunctionalTensor]
394 and t not in [torch.Tensor, FunctionalTensor]
410 if func in FunctionalTensor.maybe_aliasing_or_mutating_ops:
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | functional_utils.py | 17 from torch._subclasses.functional_tensor import FunctionalTensor
41 return FunctionalTensor.to_functional(t)
67 if not isinstance(t, FunctionalTensor):
89 return isinstance(t, FunctionalTensor)
103 assert isinstance(t, FunctionalTensor)
116 assert isinstance(t, FunctionalTensor)
130 assert isinstance(t, FunctionalTensor)
146 assert isinstance(t, FunctionalTensor)
176 assert isinstance(f_arg, FunctionalTensor)
|
| H A D | collect_metadata_analysis.py | 22 from torch._subclasses.functional_tensor import FunctionalTensor, FunctionalTensorMode
359 o, FunctionalTensor
396 o, FunctionalTensor
621 if isinstance(o, FunctionalTensor):
|
| H A D | input_output_analysis.py | 19 from torch._subclasses.functional_tensor import FunctionalTensor
198 raw_type=FunctionalTensor,
|
| H A D | traced_function_transforms.py | 550 FunctionalTensor,
553 assert isinstance(inpt_f, FunctionalTensor)
|
| H A D | dispatch_and_compile_graph.py | 134 buffer, torch._subclasses.functional_tensor.FunctionalTensor
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | __init__.py | 114 from torch._subclasses.functional_tensor import FunctionalTensor
142 for op in FunctionalTensor.maybe_aliasing_or_mutating_ops
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | utils.py | 218 from torch._subclasses.functional_tensor import FunctionalTensor
232 if isinstance(t, FunctionalTensor):
|
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | graph.py | 547 torch._subclasses.functional_tensor.FunctionalTensor,
562 torch._subclasses.functional_tensor.FunctionalTensor,
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | eager_transforms.py | 32 from torch._subclasses.functional_tensor import FunctionalTensor
1465 out = FunctionalTensor(wrapped)
1487 if isinstance(maybe_tensor, FunctionalTensor):
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _python_dispatch.py | 660 from torch._subclasses.functional_tensor import FunctionalTensor
662 if not isinstance(mutated_args[0], FunctionalTensor):
|
| H A D | flop_counter.py | 272 from torch._subclasses.functional_tensor import FunctionalTensor
273 if not isinstance(offsets, (FakeTensor, FunctionalTensor)):
|
| H A D | checkpoint.py | 1267 } | set(torch._subclasses.functional_tensor.FunctionalTensor.metadata_fns)
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | exported_program.py | 57 from torch._subclasses.functional_tensor import FunctionalTensor
204 if op_overload in FunctionalTensor.maybe_aliasing_or_mutating_ops:
208 if op_overload in FunctionalTensor.metadata_fns:
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _utils.py | 879 from torch._subclasses.functional_tensor import FunctionalTensor
881 if isinstance(t, FunctionalTensor):
|
| H A D | _tensor.py | 419 isinstance(self, torch._subclasses.functional_tensor.FunctionalTensor)
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | pass_base.py | 38 from torch._subclasses.functional_tensor import FunctionalTensor, FunctionalTensorMode
87 if any(isinstance(x, FunctionalTensor) for x in flat_xs)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_functionalization.py | 13 FunctionalTensor,
2118 x_wrapped = FunctionalTensor.to_functional(x)
|
| H A D | test_proxy_tensor.py | 14 from torch._subclasses.functional_tensor import FunctionalTensor, FunctionalTensorMode
455 x_unwrapped = FunctionalTensor.to_functional(x)
480 x_unwrapped = FunctionalTensor.to_functional(x)
|
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A D | ops.py | 521 FunctionalTensor,
525 if isinstance(new_offsets, (FakeTensor, FunctionalTensor)):
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | builder.py | 439 torch._subclasses.functional_tensor.FunctionalTensor,
1525 torch._subclasses.functional_tensor.FunctionalTensor,
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 14 FunctionalTensor,
38 return FunctionalTensor.to_functional(t)
43 if not isinstance(t, FunctionalTensor):
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_triton_kernels.py | 174 FunctionalTensor,
181 x_func = FunctionalTensor.to_functional(x)
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_repros.py | 5199 FunctionalTensor,
5214 inp = FunctionalTensor.to_functional(inp)
5218 tmp = FunctionalTensor.to_functional(tmp)
|