Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
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()
DResizeCommon.h35 std::optional<MemoryFormat> optional_memory_format) { in resize_named_tensor_()
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 …]
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()
/external/pytorch/aten/src/ATen/native/nested/
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()
/external/pytorch/aten/src/ATen/native/quantized/
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()
DQTensor.cpp191 std::optional<c10::MemoryFormat> optional_memory_format) { in quantized_clone()
/external/pytorch/aten/src/ATen/native/mkldnn/
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()
DTensorFactories.cpp15 …<Device> device, std::optional<bool> pin_memory, std::optional<c10::MemoryFormat> optional_memory_… in empty_mkldnn()
/external/pytorch/torch/csrc/autograd/
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_()
DTraceTypeManual.cpp53 std::optional<MemoryFormat> optional_memory_format) { in resize_()
72 std::optional<MemoryFormat> optional_memory_format) { in resize_as_()
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DClone.cpp19 std::optional<c10::MemoryFormat> optional_memory_format) { in clone()
/external/pytorch/aten/src/ATen/native/cuda/
DResize.cpp52 std::optional<MemoryFormat> optional_memory_format) { in resize_cuda_()
/external/pytorch/aten/src/ATen/test/
Dextension_backend_test.cpp19 …al<Device> device, std::optional<bool> pin_memory, std::optional<MemoryFormat> optional_memory_for… in empty_override()
Dvulkan_api_test.cpp215 …oid clone_test(const std::vector<int64_t>& size, std::optional<at::MemoryFormat> optional_memory_f… in clone_test()
/external/pytorch/aten/src/ATen/native/sparse/
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()
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()
/external/pytorch/aten/src/ATen/native/mps/
DTensorFactory.cpp108 std::optional<MemoryFormat> optional_memory_format) { in resize_mps_()
/external/pytorch/tools/autograd/templates/
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()
/external/pytorch/test/cpp_extensions/
Dmaia_extension.cpp24 … std::optional<bool> pin_memory, std::optional<c10::MemoryFormat> optional_memory_format) { in empty_override()
Dopen_registration_extension.cpp422 std::optional<at::MemoryFormat> optional_memory_format) { in custom_resize_()
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesViews.cpp152 std::optional<MemoryFormat> optional_memory_format) { in resize__plumbing()
/external/pytorch/torch/csrc/jit/runtime/static/
Dops.cpp1088 const auto& optional_memory_format = in __anon5bf8650b2f02() local