Home
last modified time | relevance | path

Searched refs:optional_memory_format (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DResize.cpp139 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 DTensorFactories.cpp264 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 DResizeCommon.h35 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 DTensorConversions.cpp237 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 DTensorConversions.h18 std::optional<c10::MemoryFormat> optional_memory_format);
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DTensorFactories.cpp23 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 DQTensor.cpp191 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 DNestedTensorFactories.cpp14 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 DVariableTypeManual.cpp243 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 DTraceTypeManual.cpp53 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 DResize.cpp52 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 DSparseTensor.cpp242 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 DSparseCsrTensor.cpp680 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 DTensorShape.cpp29 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 DTensorFactories.cpp15 … 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 DTensorFactory.cpp108 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 DTensorOperators.cpp84 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 Dpython_variable_methods.cpp400 …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 DClone.cpp19 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 DBatchRulesViews.cpp152 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 DTensorOptions.h438 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 Dopen_registration_extension.cpp422 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 Dmaia_extension.cpp24 … 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 Dextension_backend_test.cpp19 …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 Dops.cpp1088 const auto& optional_memory_format = in __anon11f46a8b2f02() local
1091 optional_memory_format.value_or(c10::MemoryFormat::Preserve); in __anon11f46a8b2f02()

12