/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | Resize.cpp | 139 std::optional<MemoryFormat> optional_memory_format) { in resize_as_() argument 142 !optional_memory_format.has_value(), in resize_as_() 144 optional_memory_format.value()); in resize_as_() 148 if (optional_memory_format.has_value()) { in resize_as_() 149 auto memory_format = optional_memory_format.value(); in resize_as_() 238 std::optional<MemoryFormat> optional_memory_format) { in _resize_() argument 243 if (optional_memory_format.has_value()) { in _resize_() 245 optional_memory_format.value(); in _resize_() 262 std::optional<MemoryFormat> optional_memory_format) { in resize_() argument 264 return resize_named_tensor_(self, size, optional_memory_format); in resize_() [all …]
|
H A D | TensorFactories.cpp | 264 std::optional<MemoryFormat> optional_memory_format) { in empty_names() argument 269 return at::empty(size, options, optional_memory_format); in empty_names() 275 auto result = at::empty(size, options, optional_memory_format); in empty_names() 330 std::optional<c10::MemoryFormat> optional_memory_format, in empty_out() argument 335 !optional_memory_format.has_value(), in empty_out() 376 std::optional<c10::MemoryFormat> optional_memory_format) { in AT_FORALL_SCALAR_TYPES_AND3() 383 .merge_memory_format(optional_memory_format); in AT_FORALL_SCALAR_TYPES_AND3() 387 optional_memory_format.has_value()), in AT_FORALL_SCALAR_TYPES_AND3() 429 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_like_quantized() argument 434 !(options_.has_memory_format() && optional_memory_format.has_value()), in empty_like_quantized() [all …]
|
H A D | ResizeCommon.h | 35 std::optional<MemoryFormat> optional_memory_format) { in resize_named_tensor_() argument 49 !optional_memory_format.has_value(), in resize_named_tensor_() 51 optional_memory_format.value()); in resize_named_tensor_()
|
H A D | TensorConversions.cpp | 237 std::optional<c10::MemoryFormat> optional_memory_format) { in _to_copy() argument 253 auto memory_format = optional_memory_format.value_or(MemoryFormat::Preserve); in _to_copy() 406 std::optional<c10::MemoryFormat> optional_memory_format) { in to_will_alias() argument 407 auto memory_format = optional_memory_format.value_or(MemoryFormat::Preserve); in to_will_alias() 425 std::optional<c10::MemoryFormat> optional_memory_format) { in to_impl() argument 428 if (to_will_alias(self, dtype, layout, device, copy, optional_memory_format)) { in to_impl() 432 self, dtype, layout, device, pin_memory, non_blocking, optional_memory_format); in to_impl() 480 std::optional<c10::MemoryFormat> optional_memory_format in to() argument 490 optional_memory_format); in to() 493 …ype dtype, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_format) { in to() argument [all …]
|
H A D | TensorConversions.h | 18 std::optional<c10::MemoryFormat> optional_memory_format);
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/ |
H A D | TensorFactories.cpp | 23 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_affine_quantized() argument 28 !(options_.has_memory_format() && optional_memory_format.has_value()), in empty_affine_quantized() 31 auto options = options_.merge_memory_format(optional_memory_format); in empty_affine_quantized() 51 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_per_channel_affine_quantized() argument 56 !(options_.has_memory_format() && optional_memory_format.has_value()), in empty_per_channel_affine_quantized() 59 auto options = options_.merge_memory_format(optional_memory_format); in empty_per_channel_affine_quantized() 77 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_unknown_quantized() argument 82 !(options_.has_memory_format() && optional_memory_format.has_value()), in empty_unknown_quantized() 85 auto options = options_.merge_memory_format(optional_memory_format); in empty_unknown_quantized()
|
H A D | QTensor.cpp | 191 std::optional<c10::MemoryFormat> optional_memory_format) { in quantized_clone() argument 193 optional_memory_format.value_or(MemoryFormat::Contiguous); in quantized_clone()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
H A D | NestedTensorFactories.cpp | 14 std::optional<c10::MemoryFormat> optional_memory_format) { in verify_empty_parameters() argument 20 .memory_format(optional_memory_format); in verify_empty_parameters() 32 !(options.layout() != kStrided && optional_memory_format.has_value()), in verify_empty_parameters() 43 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_like_nested() argument 45 self, dtype, layout, device, pin_memory, optional_memory_format); in empty_like_nested() 90 std::optional<c10::MemoryFormat> optional_memory_format) { in _to_copy_nested() argument 107 auto memory_format = optional_memory_format.value_or(MemoryFormat::Preserve); in _to_copy_nested() 134 std::optional<c10::MemoryFormat> optional_memory_format) { in clone_nested() argument 135 auto memory_format = optional_memory_format.value_or(c10::MemoryFormat::Preserve); in clone_nested()
|
/aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
H A D | VariableTypeManual.cpp | 243 std::optional<MemoryFormat> optional_memory_format) { in resize_() argument 251 ks & c10::after_autograd_keyset, self_, size, optional_memory_format); in resize_() 265 std::optional<MemoryFormat> optional_memory_format) { in resize_as_() argument 277 optional_memory_format); in resize_as_() 402 std::optional<MemoryFormat> optional_memory_format) { in resize_() argument 413 optional_memory_format); in resize_() 426 std::optional<MemoryFormat> optional_memory_format) { in resize_as_() argument 437 optional_memory_format); in resize_as_()
|
H A D | TraceTypeManual.cpp | 53 std::optional<MemoryFormat> optional_memory_format) { in resize_() argument 64 self.resize_(size, optional_memory_format); in resize_() 72 std::optional<MemoryFormat> optional_memory_format) { in resize_as_() argument 80 self.resize_as_(the_template, optional_memory_format); in resize_as_()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | Resize.cpp | 52 std::optional<MemoryFormat> optional_memory_format) { in resize_cuda_() argument 54 return resize_named_tensor_(self, size, optional_memory_format); in resize_cuda_() 59 if (optional_memory_format.has_value()) { in resize_cuda_() 61 optional_memory_format.value(); in resize_cuda_()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
H A D | SparseTensor.cpp | 242 std::optional<MemoryFormat> optional_memory_format) { in empty_sparse_symint() argument 244 …se(C10_AS_INTARRAYREF_SLOW_ALLOC(size), dtype, layout, device, pin_memory, optional_memory_format); in empty_sparse_symint() 253 std::optional<MemoryFormat> optional_memory_format) { in empty_sparse() argument 523 std::optional<c10::MemoryFormat> optional_memory_format) { in clone_sparse() argument 525 !optional_memory_format.has_value(), in clone_sparse() 527 optional_memory_format.value()); in clone_sparse() 846 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_like_sparse_coo() argument 850 !(options_.has_memory_format() && optional_memory_format.has_value()), in empty_like_sparse_coo() 857 .merge_memory_format(optional_memory_format); in empty_like_sparse_coo() 861 optional_memory_format.has_value()), in empty_like_sparse_coo() [all …]
|
H A D | SparseCsrTensor.cpp | 680 std::optional<MemoryFormat> optional_memory_format) { in SPARSE_COMPRESSED_TENSOR() 682 …ed(C10_AS_INTARRAYREF_SLOW_ALLOC(size), dtype, layout, device, pin_memory, optional_memory_format); in SPARSE_COMPRESSED_TENSOR() 695 std::optional<MemoryFormat> optional_memory_format) { in empty_sparse_compressed() argument 740 std::optional<MemoryFormat> optional_memory_format) { in resize_sparse_csr_() argument 877 std::optional<c10::MemoryFormat> optional_memory_format) { in clone_sparse_compressed() argument 879 !optional_memory_format.has_value(), in clone_sparse_compressed() 881 optional_memory_format.value()); in clone_sparse_compressed() 908 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_like_sparse_csr() argument 913 .merge_memory_format(optional_memory_format); in empty_like_sparse_csr() 960 return at::native::empty_like(self, dtype, layout, device, pin_memory, optional_memory_format); in empty_like_sparse_csr()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
H A D | TensorShape.cpp | 29 Tensor mkldnn_clone(const Tensor& self, std::optional<c10::MemoryFormat> optional_memory_format) { in mkldnn_clone() argument 68 Tensor mkldnn_clone(const Tensor& self, std::optional<c10::MemoryFormat> optional_memory_format) { in mkldnn_clone() argument 70 !optional_memory_format.has_value(), in mkldnn_clone() 72 optional_memory_format.value()); in mkldnn_clone()
|
H A D | TensorFactories.cpp | 15 … device, std::optional<bool> pin_memory, std::optional<c10::MemoryFormat> optional_memory_format) { in empty_mkldnn() argument 17 !optional_memory_format.has_value(), in empty_mkldnn() 29 … device, std::optional<bool> pin_memory, std::optional<c10::MemoryFormat> optional_memory_format) {
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/ |
H A D | TensorFactory.cpp | 108 std::optional<MemoryFormat> optional_memory_format) { in resize_mps_() argument 110 return resize_named_tensor_(self, size, optional_memory_format); in resize_mps_() 115 if (optional_memory_format.has_value()) { in resize_mps_() 117 optional_memory_format.value(); in resize_mps_()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
H A D | TensorOperators.cpp | 84 std::optional<MemoryFormat> optional_memory_format) { in AT_FORALL_OPERATORS() 89 !optional_memory_format.has_value(), in AT_FORALL_OPERATORS() 91 optional_memory_format.value()); in AT_FORALL_OPERATORS()
|
/aosp_15_r20/external/pytorch/tools/autograd/templates/ |
H A D | python_variable_methods.cpp | 400 …ce device, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_format) { in dispatch_to() argument 407 …return self.to(self.options().device(device).memory_format(optional_memory_format), non_blocking, … in dispatch_to() 410 …or & self, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_format) { in dispatch_to() argument 412 return self.to(self.options().memory_format(optional_memory_format), non_blocking, copy); in dispatch_to() 415 …ype dtype, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_format) { in dispatch_to() argument 418 return self.to(dtype, non_blocking, copy, optional_memory_format); in dispatch_to() 421 …ype dtype, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_format) { in dispatch_to() argument 424 return self.to(device, dtype, non_blocking, copy, optional_memory_format); in dispatch_to() 572 …e(PyObject* self, ScalarType scalarType, std::optional<c10::MemoryFormat> optional_memory_format) { in THPVariable_to_type() argument 575 return THPVariable_Wrap(dispatch_to(self_, scalarType, false, false, optional_memory_format)); in THPVariable_to_type()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
H A D | Clone.cpp | 19 std::optional<c10::MemoryFormat> optional_memory_format) { in clone() argument 20 auto memory_format = optional_memory_format.value_or(MemoryFormat::Preserve); in clone()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
H A D | BatchRulesViews.cpp | 152 std::optional<MemoryFormat> optional_memory_format) { in resize__plumbing() argument 154 !optional_memory_format.has_value() || in resize__plumbing() 155 optional_memory_format == c10::MemoryFormat::Contiguous, in resize__plumbing() 162 return self.resize_(size, optional_memory_format); in resize__plumbing()
|
/aosp_15_r20/external/pytorch/c10/core/ |
H A D | TensorOptions.h | 438 std::optional<MemoryFormat> optional_memory_format) const noexcept { in merge_memory_format() 440 if (optional_memory_format.has_value()) { in merge_memory_format() 441 merged.set_memory_format(*optional_memory_format); in merge_memory_format()
|
/aosp_15_r20/external/pytorch/test/cpp_extensions/ |
H A D | open_registration_extension.cpp | 422 std::optional<at::MemoryFormat> optional_memory_format) { in custom_resize_() argument 432 if (optional_memory_format.has_value()) { in custom_resize_() 434 optional_memory_format.value(); in custom_resize_()
|
H A D | maia_extension.cpp | 24 … std::optional<bool> pin_memory, std::optional<c10::MemoryFormat> optional_memory_format) { in empty_override() argument
|
/aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
H A D | extension_backend_test.cpp | 19 …vice> device, std::optional<bool> pin_memory, std::optional<MemoryFormat> optional_memory_format) { in empty_override() argument
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
H A D | ops.cpp | 1088 const auto& optional_memory_format = in __anon11f46a8b2f02() local 1091 optional_memory_format.value_or(c10::MemoryFormat::Preserve); in __anon11f46a8b2f02()
|