Home
last modified time | relevance | path

Searched defs:device_opt (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DEmptyTensor.cpp11 std::optional<Device> device_opt, in empty_cuda()
27 std::optional<Device> device_opt, in empty_cuda()
52 std::optional<Device> device_opt) { in empty_strided_cuda()
68 std::optional<Device> device_opt, in empty_strided_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DEmptyTensor.cpp267 std::optional<Device> device_opt, in empty_cpu()
302 std::optional<Device> device_opt, in empty_strided_cpu()
358 std::optional<Device> device_opt, in empty_meta()
378 std::optional<Device> device_opt, in empty_symint_meta()
412 std::optional<Device> device_opt, in empty_strided_meta()
447 std::optional<Device> device_opt) { in empty_strided_symint_meta()
H A DScalarOps.cpp26 …or_static(const Scalar& s, std::optional<ScalarType> dtype_opt, std::optional<Device> device_opt) { in scalar_tensor_static()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorFactories.cu54 …al<ScalarType> dtype_opt, std::optional<Layout> layout_opt, std::optional<Device> device_opt, std:… in empty_cuda()
80 …al<ScalarType> dtype_opt, std::optional<Layout> layout_opt, std::optional<Device> device_opt, std:… in empty_strided_cuda()
278 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in tril_indices_cuda()
354 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in triu_indices_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DMetaTensor.cpp18 std::optional<Device> device_opt, in empty_meta_symint()
36 std::optional<Device> device_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 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in empty_permuted_symint()
320 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in empty_strided_cpu()
505 std::optional<Device> device_opt, in new_empty_symint()
1181 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in tril_indices_cpu()
1231 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in triu_indices_cpu()
H A DConstraints.cpp83 std::optional<Device> device_opt, in _make_dep_token_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DEmptyTensor.cpp25 std::optional<Device> device_opt, in empty_mps()
98 std::optional<Device> device_opt) { in empty_strided_mps()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DTensorFactory.cpp79 std::optional<Device> device_opt, in empty_mps()
91 std::optional<Device> device_opt, in empty_strided_mps()
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegMem.cpp65 std::optional<c10::Device> device_opt, in empty_openreg()
84 std::optional<c10::Device> device_opt, in empty_strided_openreg()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DStorage.cpp321 std::optional<at::Device> device_opt = r.deviceOptional(device_arg_idx); in THPStorage_pynew() local
503 std::optional<at::Device> device_opt = old_storage_impl->device(); in THPStorage_get() local
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.cpp130 std::optional<c10::Device> device_opt) in TensorImpl()
138 std::optional<c10::Device> device_opt) in TensorImpl()
H A DStorageImpl.cpp75 std::optional<at::Device> device_opt) { in make_storage_impl()
H A DTensorOptions.h289 std::optional<Device> device_opt() const noexcept { in device_opt() function
H A DTensorImpl.h2138 std::optional<c10::Device> device_opt() const { in device_opt() function
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DDeprecatedTypeProperties.h110 TensorOptions options(std::optional<Device> device_opt) const { in options()
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DInlineDeviceGuard.h226 explicit InlineOptionalDeviceGuard(std::optional<Device> device_opt) in InlineOptionalDeviceGuard()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcodegen.h87 std::optional<c10::Device> device_opt, in empty_strided()
H A Dcuda_codegen.cpp1255 std::optional<c10::Device> device_opt, in empty_strided()
H A Dllvm_codegen.cpp451 std::optional<c10::Device> device_opt, in empty_strided()
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_device.cpp137 …larType> dtype_opt, std::optional<at::Layout> layout_opt, std::optional<at::Device> device_opt, st… in custom_empty_strided()
/aosp_15_r20/device/google/cuttlefish/host/frontend/webrtc/html_client/js/
Dcf_webrtc.js348 async #useDevice(in_use, senders_arr, device_opt, requestedFn = () => {in_use}, argument
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp388 std::optional<at::Device> device_opt, in custom_empty_strided()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.cpp293 std::optional<Device> device_opt, in internal_new_from_data()
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py6451 device_opt, argument