| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorFactories.cpp | 390 auto memory_format = options.memory_format_opt().value_or(MemoryFormat::Preserve); in AT_FORALL_SCALAR_TYPES_AND3() local
448 auto memory_format = options.memory_format_opt().value_or(MemoryFormat::Preserve); in empty_like_quantized() local
1740 auto memory_format = in clone() local
|
| H A D | TensorConversions.cpp | 253 auto memory_format = optional_memory_format.value_or(MemoryFormat::Preserve); in _to_copy() local
407 auto memory_format = optional_memory_format.value_or(MemoryFormat::Preserve); in to_will_alias() local
|
| H A D | Convolution.cpp | 830 const auto memory_format = tensor.suggest_memory_format(); in subtensor() local
|
| H A D | TensorShape.cpp | 272 auto memory_format = cat_compute_output_memory_format(materialized); in TORCH_PRECOMPUTE_META_FUNC() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/quantized/ |
| H A D | Quantizer.cpp | 112 auto memory_format = options.memory_format_opt().value_or(MemoryFormat::Contiguous); in new_qtensor() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 1083 ::std::optional<at::MemoryFormat> memory_format) { in compute_shape__to_copy()
1092 ::std::optional<at::MemoryFormat> memory_format) { in compute_shape_clone()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | TensorImpl.cpp | 922 void TensorImpl::empty_tensor_restride_symint(MemoryFormat memory_format) { in empty_tensor_restride_symint()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Shape.cu | 289 int nDims, c10::MemoryFormat memory_format) { in parallel_cat()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 2311 def helper(size, memory_format): argument
6307 def helper(N, C, H, W, memory_format): argument
6689 def helper(shape, alpha=1.0, memory_format=torch.contiguous_format): argument
10285 def test_helper(shape, memory_format, device="mps"): argument
10295 def test_helper(shape, numel, memory_format, device="mps"): argument
10537 def helper(m_cpu, memory_format): argument
|
| H A D | test_legacy_vmap.py | 1780 def baz(x, memory_format): argument
|
| H A D | test_reductions.py | 1410 def _test_memory_format_transformations(self, device, input_generator_fn, transformation_fn, argument
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | copy_ops_util.cpp | 733 exec_aten::optional<exec_aten::MemoryFormat> memory_format, in check_to_copy_args()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops.cpp | 636 auto memory_format = pop(stack).toMemoryFormat(); in __anonbfe5918f2b02() local
647 auto memory_format = pop(stack).toMemoryFormat(); in __anonbfe5918f2c02() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_arg_parser.h | 895 const auto memory_format = reinterpret_cast<THPMemoryFormat*>(args[i]); in memoryformat() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | tensor.py | 610 def method_is_contiguous(self, memory_format=None): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 4204 self, model, input_var, target, loss, scale_factor=1.0, memory_format=None argument
4232 memory_format=None, argument
5845 def _test_DistributedDataParallel_SyncBatchNorm_with_memory_format( argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_sugared_value.cpp | 1168 auto memory_format = reinterpret_cast<THPMemoryFormat*>(obj.ptr()); in toSugaredValue() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 693 memory_format=None, argument
3791 memory_format, argument
4770 memory_format=None, argument
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | __init__.py | 683 input: TensorLikeType, *, memory_format: torch.memory_format = torch.preserve_format argument
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 2158 memory_format: Optional[torch.memory_format] = None, argument
5081 def resize_as(self, other, memory_format=None): argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 3505 def fn(a, memory_format): argument
9479 self, fn, x, size_or_y, memory_format, inplace, deterministic argument
9534 def fn(x, size, memory_format): argument
9569 def fn(x, y, memory_format): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | external_functions.cpp | 51 auto memory_format = deduce_memory_format(strides, sizes); in from_blob_quantized() local
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 2455 def baz(x, memory_format): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 2813 def stride_ordered_for_memory_format(sizes, memory_format): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 2196 const auto memory_format = indices.suggest_memory_format(); in max_pool_double_backward() local
|