| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorFactories.cpp | 139 std::optional<bool> pin_memory) { in arange()
147 std::optional<bool> pin_memory) { in arange()
159 std::optional<bool> pin_memory) { in arange()
263 std::optional<bool> pin_memory, in empty_names()
428 std::optional<bool> pin_memory, in empty_like_quantized()
511 …auto pin_memory = pin_memory_opt.has_value() ? pin_memory_opt : self.options().pinned_memory_opt(); in new_empty_symint() local
522 std::optional<bool> pin_memory in new_empty_strided_symint()
536 std::optional<bool> pin_memory) { in eye()
545 std::optional<bool> pin_memory) { in eye()
612 std::optional<bool> pin_memory) { in full()
[all …]
|
| H A D | Memory.cpp | 44 Tensor pin_memory(const Tensor& self, std::optional<c10::Device> device) { in pin_memory() function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset8.py | 355 pin_memory=False, argument
369 pin_memory=False, argument
377 def zeros(g: jit_utils.GraphContext, sizes, dtype, layout, device, pin_memory=False): argument
390 pin_memory=False, argument
399 def ones(g: jit_utils.GraphContext, sizes, dtype, layout, device, pin_memory=False): argument
411 pin_memory=False, argument
420 g: jit_utils.GraphContext, sizes, value, dtype, layout, device, pin_memory=False argument
440 pin_memory=False, argument
|
| H A D | symbolic_opset9.py | 3415 pin_memory=False, argument
3429 pin_memory=False, argument
3437 g: jit_utils.GraphContext, self, sizes, dtype, layout, device, pin_memory=False argument
3489 def zeros(g: jit_utils.GraphContext, sizes, dtype, layout, device, pin_memory=False): argument
3513 pin_memory=False, argument
3532 g: jit_utils.GraphContext, self, sizes, dtype, layout, device, pin_memory=False argument
3549 def ones(g: jit_utils.GraphContext, sizes, dtype, layout, device, pin_memory=False): argument
3572 pin_memory=False, argument
3591 g: jit_utils.GraphContext, self, sizes, dtype, layout, device, pin_memory=False argument
3601 g: jit_utils.GraphContext, sizes, value, dtype, layout, device, pin_memory=False argument
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensor.cpp | 152 std::optional<bool> pin_memory) { in new_sparse()
179 std::optional<bool> pin_memory) { in new_with_dims_sparse()
194 std::optional<bool> pin_memory, in new_with_dims_and_tensor_sparse_symint()
241 std::optional<bool> pin_memory, in empty_sparse_symint()
252 std::optional<bool> pin_memory, in empty_sparse()
266 std::optional<bool> pin_memory) { in sparse_coo_tensor()
292 std::optional<bool> pin_memory, in sparse_coo_tensor()
457 std::optional<bool> pin_memory, in sparse_coo_tensor()
481 std::optional<bool> pin_memory, in _sparse_coo_tensor_unsafe()
499 std::optional<bool> pin_memory, in _sparse_coo_tensor_unsafe_symint()
[all …]
|
| H A D | SparseCsrTensor.cpp | 385 std::optional<bool> pin_memory) { in sparse_compressed_tensor_with_dims()
460 std::optional<bool> pin_memory) { in _sparse_compressed_tensor_unsafe_symint()
487 std::optional<bool> pin_memory) { in _sparse_compressed_tensor_unsafe_template()
587 std::optional<bool> pin_memory) { in sparse_compressed_tensor()
616 std::optional<bool> pin_memory) { in sparse_compressed_tensor()
694 std::optional<bool> pin_memory, in empty_sparse_compressed()
907 std::optional<bool> pin_memory, in empty_like_sparse_csr()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesFactory.cpp | 114 std::optional<bool> pin_memory) in linspace_logspace_batch_rule_helper()
151 std::optional<bool> pin_memory){ in linspace_Tensor_Tensor_batch_rule()
162 std::optional<bool> pin_memory){ in linspace_Tensor_Scalar_batch_rule()
175 std::optional<bool> pin_memory){ in linspace_Scalar_Tensor_batch_rule()
189 std::optional<bool> pin_memory){ in logspace_Tensor_Tensor_batch_rule()
201 std::optional<bool> pin_memory){ in logspace_Tensor_Scalar_batch_rule()
215 std::optional<bool> pin_memory){ in logspace_Scalar_Tensor_batch_rule()
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| H A D | __init__.py | 32 pin_memory=False, argument
87 pin_memory=False, argument
141 pin_memory=False, argument
197 pin_memory=False, argument
251 pin_memory=False, argument
307 pin_memory=False, argument
|
| H A D | metadata.py | 25 pin_memory: bool = False variable in TensorProperties
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/ |
| H A D | TensorFactories.cpp | 20 std::optional<bool> pin_memory, in empty_affine_quantized()
50 std::optional<bool> pin_memory, in empty_per_channel_affine_quantized()
76 std::optional<bool> pin_memory, in empty_unknown_quantized()
99 std::optional<bool> pin_memory) { in empty_strided_unknown_quantized()
139 std::optional<bool> pin_memory, in empty_quantized()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Factory.cpp | 14 const std::optional<bool> pin_memory, in _empty_affine_quantized()
36 const std::optional<bool> pin_memory, in empty_memory_format()
55 const std::optional<bool> pin_memory) { in empty_strided()
|
| H A D | Zero.cpp | 49 std::optional<bool> pin_memory) { in zeros()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ops/ |
| H A D | to_copy.h | 24 const std::optional<bool>& pin_memory, in ToCopy()
53 const std::optional<bool>& pin_memory, in CanBeReused()
121 std::optional<bool> pin_memory; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | EmptyTensor.cpp | 14 c10::Allocator* GetCPUAllocatorMaybePinned(bool pin_memory) { in GetCPUAllocatorMaybePinned()
256 TensorBase empty_cpu(IntArrayRef size, ScalarType dtype, bool pin_memory, in empty_cpu()
273 auto pin_memory = pinned_memory_or_default(pin_memory_opt); in empty_cpu() local
290 ScalarType dtype, bool pin_memory) { in empty_strided_cpu()
307 auto pin_memory = pinned_memory_or_default(pin_memory_opt); in empty_strided_cpu() local
|
| H A D | LegacyBatchingRegistrations.cpp | 964 std::optional<bool> pin_memory, in to_dtype_layout_batching_rule()
984 std::optional<bool> pin_memory) { in new_zeros_batching_rule()
1002 std::optional<bool> pin_memory) { in new_empty_batching_rule()
1016 std::optional<bool> pin_memory) { in new_empty_strided_batching_rule()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_native_functions.cpp | 145 std::optional<bool> pin_memory, in _to_copy()
277 std::optional<bool> pin_memory, in empty_symint()
308 std::optional<bool> pin_memory) { in empty_strided_symint()
413 std::optional<bool> pin_memory) { in new_empty_strided_symint()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorFactories.cpp | 13 std::optional<bool> pin_memory, in verify_empty_parameters()
42 std::optional<bool> pin_memory, in empty_like_nested()
88 std::optional<bool> pin_memory, in _to_copy_nested()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | extension_backend_test.cpp | 19 …std::optional<Device> device, std::optional<bool> pin_memory, std::optional<MemoryFormat> optional… in empty_override()
45 std::optional<bool> pin_memory) { in empty_strided_override()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_new.cpp | 298 bool pin_memory = false) { in internal_new_from_data()
949 const bool pin_memory = r.toBool(ARG_PIN_MEMORY); in sparse_compressed_tensor_ctor_worker() local
1011 const bool pin_memory = r.toBool(ARG_PIN_MEMORY1); in sparse_compressed_tensor_ctor_worker() local
1176 bool pin_memory = r.toBool(ARG_PIN_MEMORY); in sparse_coo_tensor_ctor() local
1210 bool pin_memory = r.toBool(ARG_PIN_MEMORY1); in sparse_coo_tensor_ctor() local
1468 bool pin_memory = r.toBool(3); in tensor_ctor() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | replace_random.py | 105 pin_memory=None, argument
138 pin_memory=None, argument
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions_for_rng.py | 33 def rand(shape, dtype=None, layout=torch.strided, device=None, pin_memory=False): argument
51 pin_memory=False, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | MiscUtils.h | 21 static inline Storage pin_memory(int64_t size) { in pin_memory() function
|
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/ |
| H A D | _fsdp_api.py | 80 pin_memory: bool = True variable in CPUOffloadPolicy
|
| /aosp_15_r20/external/pytorch/torch/utils/data/_utils/ |
| H A D | pin_memory.py | 62 def pin_memory(data, device=None): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | TensorFactories.cpp | 15 … std::optional<Layout> layout, std::optional<Device> device, std::optional<bool> pin_memory, std::… in empty_mkldnn()
|