Home
last modified time | relevance | path

Searched defs:optional_memory_format (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DResize.cpp139 std::optional<MemoryFormat> optional_memory_format) { in resize_as_()
238 std::optional<MemoryFormat> optional_memory_format) { in _resize_()
262 std::optional<MemoryFormat> optional_memory_format) { in resize_()
272 std::optional<MemoryFormat> optional_memory_format) { in resize__symint()
H A DTensorFactories.cpp264 std::optional<MemoryFormat> optional_memory_format) { in empty_names()
330 std::optional<c10::MemoryFormat> optional_memory_format, in empty_out()
429 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_like_quantized()
638 std::optional<c10::MemoryFormat> optional_memory_format) { in full_like()
822 std::optional<c10::MemoryFormat> optional_memory_format) { in ones_like()
903 std::optional<c10::MemoryFormat> optional_memory_format) { in rand_like()
991 std::optional<c10::MemoryFormat> optional_memory_format) { in randint_like()
1007 std::optional<c10::MemoryFormat> optional_memory_format) { in randint_like()
1071 std::optional<c10::MemoryFormat> optional_memory_format) { in randn_like()
1371 std::optional<c10::MemoryFormat> optional_memory_format) { in zeros_like()
[all …]
H A DTensorConversions.cpp237 std::optional<c10::MemoryFormat> optional_memory_format) { in _to_copy()
406 std::optional<c10::MemoryFormat> optional_memory_format) { in to_will_alias()
425 std::optional<c10::MemoryFormat> optional_memory_format) { in to_impl()
480 std::optional<c10::MemoryFormat> optional_memory_format in to()
493 … ScalarType dtype, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_… in to()
505 … ScalarType dtype, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_… in to()
517 …nst Tensor& other, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_… in to()
H A DResizeCommon.h35 std::optional<MemoryFormat> optional_memory_format) { in resize_named_tensor_()
/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()
43 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_like_nested()
90 std::optional<c10::MemoryFormat> optional_memory_format) { in _to_copy_nested()
134 std::optional<c10::MemoryFormat> optional_memory_format) { in clone_nested()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DVariableTypeManual.cpp243 std::optional<MemoryFormat> optional_memory_format) { in resize_()
265 std::optional<MemoryFormat> optional_memory_format) { in resize_as_()
402 std::optional<MemoryFormat> optional_memory_format) { in resize_()
426 std::optional<MemoryFormat> optional_memory_format) { in resize_as_()
H A DTraceTypeManual.cpp53 std::optional<MemoryFormat> optional_memory_format) { in resize_()
72 std::optional<MemoryFormat> optional_memory_format) { in resize_as_()
/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()
51 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_per_channel_affine_quantized()
77 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_unknown_quantized()
H A DQTensor.cpp191 std::optional<c10::MemoryFormat> optional_memory_format) { in quantized_clone()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensor.cpp242 std::optional<MemoryFormat> optional_memory_format) { in empty_sparse_symint()
253 std::optional<MemoryFormat> optional_memory_format) { in empty_sparse()
523 std::optional<c10::MemoryFormat> optional_memory_format) { in clone_sparse()
846 std::optional<c10::MemoryFormat> optional_memory_format) { in empty_like_sparse_coo()
H A DSparseCsrTensor.cpp695 std::optional<MemoryFormat> optional_memory_format) { in empty_sparse_compressed()
740 std::optional<MemoryFormat> optional_memory_format) { in resize_sparse_csr_()
877 std::optional<c10::MemoryFormat> optional_memory_format) { in clone_sparse_compressed()
908 std::optional<c10::MemoryFormat> 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()
68 Tensor mkldnn_clone(const Tensor& self, std::optional<c10::MemoryFormat> optional_memory_format) { in mkldnn_clone()
H A DTensorFactories.cpp15 …<Device> device, std::optional<bool> pin_memory, std::optional<c10::MemoryFormat> optional_memory_… in empty_mkldnn()
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp400 …lf, Device device, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_… in dispatch_to()
410 …nst Tensor & self, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_… in dispatch_to()
415 … ScalarType dtype, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_… in dispatch_to()
421 … ScalarType dtype, bool non_blocking, bool copy, std::optional<c10::MemoryFormat> optional_memory_… in dispatch_to()
572 …e_to_type(PyObject* self, ScalarType scalarType, std::optional<c10::MemoryFormat> optional_memory_… 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DResize.cpp52 std::optional<MemoryFormat> optional_memory_format) { in resize_cuda_()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dextension_backend_test.cpp19 …al<Device> device, std::optional<bool> pin_memory, std::optional<MemoryFormat> optional_memory_for… in empty_override()
H A Dvulkan_api_test.cpp215 …oid clone_test(const std::vector<int64_t>& size, std::optional<at::MemoryFormat> optional_memory_f… in clone_test()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DTensorFactory.cpp108 std::optional<MemoryFormat> optional_memory_format) { in resize_mps_()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmaia_extension.cpp24 … std::optional<bool> pin_memory, std::optional<c10::MemoryFormat> optional_memory_format) { in empty_override()
H A Dopen_registration_extension.cpp422 std::optional<at::MemoryFormat> optional_memory_format) { in custom_resize_()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesViews.cpp152 std::optional<MemoryFormat> optional_memory_format) { in resize__plumbing()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp1088 const auto& optional_memory_format = in __anon11f46a8b2f02() local