Home
last modified time | relevance | path

Searched defs:pin_memory_opt (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DEmptyTensor.cpp268 std::optional<bool> pin_memory_opt, in empty_cpu()
303 std::optional<bool> pin_memory_opt) { in empty_strided_cpu()
359 std::optional<bool> pin_memory_opt, in empty_meta()
379 std::optional<bool> pin_memory_opt, in empty_symint_meta()
413 std::optional<bool> pin_memory_opt) { in empty_strided_meta()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorFactories.cu54 …onal<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt, std… in empty_cuda()
80 …ptional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt) { in empty_strided_cuda()
278 …ptional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt) { in tril_indices_cuda()
354 …ptional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt) { in triu_indices_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DMetaTensor.cpp19 std::optional<bool> pin_memory_opt, in empty_meta_symint()
37 std::optional<bool> pin_memory_opt in empty_strided_meta_symint()
H A DTensorFactories.cpp248 …std::optional<Device> device_opt, std::optional<bool> pin_memory_opt, std::optional<c10::MemoryFor… in empty_cpu()
281 …::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt in empty_permuted_symint()
320 …ptional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt) { in empty_strided_cpu()
506 std::optional<bool> pin_memory_opt in new_empty_symint()
1181 …ptional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt) { in tril_indices_cpu()
1231 …ptional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory_opt) { in triu_indices_cpu()
H A DConstraints.cpp84 std::optional<bool> pin_memory_opt, in _make_dep_token_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DEmptyTensor.cpp28 std::optional<bool> pin_memory_opt, in empty_cuda()
69 std::optional<bool> pin_memory_opt) { in empty_strided_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DTensorFactory.cpp80 std::optional<bool> pin_memory_opt, in empty_mps()
92 std::optional<bool> pin_memory_opt) { in empty_strided_mps()
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegMem.cpp66 std::optional<bool> pin_memory_opt, in empty_openreg()
85 std::optional<bool> pin_memory_opt) { in empty_strided_openreg()
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DEmptyTensor.cpp26 std::optional<bool> pin_memory_opt, in empty_mps()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcodegen.h88 std::optional<bool> pin_memory_opt) { in empty_strided()
H A Dcuda_codegen.cpp1256 std::optional<bool> pin_memory_opt) { in empty_strided()
H A Dllvm_codegen.cpp452 std::optional<bool> pin_memory_opt) { in empty_strided()
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_device.cpp137 …at::Layout> layout_opt, std::optional<at::Device> device_opt, std::optional<bool> pin_memory_opt) { in custom_empty_strided()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp389 std::optional<bool> pin_memory_opt) { in custom_empty_strided()