Home
last modified time | relevance | path

Searched defs:memory_format_opt (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DEmptyTensor.cpp168 std::optional<c10::MemoryFormat> memory_format_opt) { in _empty_generic()
202 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_generic()
211 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_generic_symint()
257 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_cpu()
269 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_cpu()
347 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_meta()
360 std::optional<c10::MemoryFormat> memory_format_opt in empty_meta()
380 std::optional<c10::MemoryFormat> memory_format_opt in empty_symint_meta()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DEmptyTensor.cpp12 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_cuda()
29 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DMetaTensor.cpp20 std::optional<c10::MemoryFormat> memory_format_opt in empty_meta_symint()
H A DConstraints.cpp85 std::optional<c10::MemoryFormat> memory_format_opt) { in _make_dep_token_cpu()
H A DTensorFactories.cpp248 …e> device_opt, std::optional<bool> pin_memory_opt, std::optional<c10::MemoryFormat> memory_format_… in empty_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DEmptyTensor.cpp27 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_mps()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DTensorFactory.cpp81 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_mps()
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegMem.cpp67 std::optional<c10::MemoryFormat> memory_format_opt) { in empty_openreg()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorFactories.cu54 …e> device_opt, std::optional<bool> pin_memory_opt, std::optional<c10::MemoryFormat> memory_format_… in empty_cuda()
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorOptions.h397 std::optional<MemoryFormat> memory_format_opt() const noexcept { in memory_format_opt() function