| /external/pytorch/aten/src/ATen/native/ |
| 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 …]
|
| D | Memory.cpp | 44 Tensor pin_memory(const Tensor& self, std::optional<c10::Device> device) { in pin_memory() function
|
| /external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| 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
|
| D | metadata.py | 25 pin_memory: bool = False variable in TensorProperties
|
| /external/pytorch/aten/src/ATen/functorch/ |
| 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()
|
| /external/pytorch/aten/src/ATen/native/quantized/ |
| 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()
|
| /external/pytorch/torch/onnx/ |
| 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
|
| /external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| 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()
|
| D | Zero.cpp | 49 std::optional<bool> pin_memory) { in zeros()
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ops/ |
| 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
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| 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 …]
|
| 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()
|
| /external/pytorch/aten/src/ATen/test/ |
| 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()
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | replace_random.py | 105 pin_memory=None, argument
138 pin_memory=None, argument
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | MiscUtils.h | 21 static inline Storage pin_memory(int64_t size) { in pin_memory() function
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| 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()
|
| /external/pytorch/torch/utils/data/_utils/ |
| D | pin_memory.py | 62 def pin_memory(data, device=None): function
|
| /external/pytorch/torch/distributed/_composable/fsdp/ |
| D | _fsdp_api.py | 80 pin_memory: bool = True variable in CPUOffloadPolicy
|
| /external/pytorch/aten/src/ATen/ |
| 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
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | TensorFactories.cpp | 15 … std::optional<Layout> layout, std::optional<Device> device, std::optional<bool> pin_memory, std::… in empty_mkldnn()
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ |
| 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()
|
| /external/pytorch/torch/csrc/utils/ |
| D | nested.cpp | 41 bool pin_memory = r.toBool(3); in nested_tensor_ctor() local
|
| /external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| D | rnn.h | 101 PackedSequence pin_memory() const { in pin_memory() function
|
| /external/pytorch/torch/_decomp/ |
| 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
|
| /external/pytorch/torch/nested/ |
| D | __init__.py | 194 …or_list, *, dtype=None, layout=None, device=None, requires_grad=False, pin_memory=False) -> Tensor: argument
|