Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/pytorch/torch/csrc/xpu/
H A DModule.cpp52 auto device_index = THPUtils_unpackDeviceIndex(arg); in THXPModule_setDevice_wrap() local
63 auto device_index = THPUtils_unpackDeviceIndex(arg); in THXPModule_exchangeDevice_wrap() local
80 auto device_index = THPUtils_unpackDeviceIndex(arg); in THXPModule_maybeExchangeDevice_wrap() local
95 auto device_index = c10::xpu::current_device(); in THXPModule_getDevice_wrap() local
110 PyObject* device_index) { in THXPModule_getCurrentStream_wrap()
131 PyObject* device_index) { in THXPModule_getCurrentStream_raw()
148 int64_t device_index = 0; in THXPModule_setStream_wrap() local
182 auto device_index = THPUtils_unpackDeviceIndex(arg); in THXPModule_xpuSynchronize() local
203 const auto device_index = THPUtils_unpackDeviceIndex(arg); in THXPModule_memoryStats() local
247 const auto device_index = THPUtils_unpackDeviceIndex(arg); in THXPModule_resetPeakMemoryStats() local
[all …]
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAStream.cpp202 static void initSingleStream(int p, DeviceIndex device_index, int i) { in initSingleStream()
217 static void initDeviceStreamState(DeviceIndex device_index) { in initDeviceStreamState()
246 static inline void check_gpu(DeviceIndex device_index) { in check_gpu()
257 CUDAStream CUDAStreamForId(DeviceIndex device_index, StreamId stream_id) { in CUDAStreamForId()
270 c10::DeviceIndex device_index = stream_.device_index(); in stream() local
315 CUDAStream getStreamFromPool(const int priority, DeviceIndex device_index) { in getStreamFromPool()
348 DeviceIndex device_index) { in getStreamFromExternal()
353 CUDAStream getDefaultCUDAStream(DeviceIndex device_index) { in getDefaultCUDAStream()
363 CUDAStream getCurrentCUDAStream(DeviceIndex device_index) { in getCurrentCUDAStream()
H A DCUDAFunctions.cpp162 for (const auto device_index : c10::irange(at::cuda::device_count())) { in getDeviceIndexWithPrimaryContext() local
173 bool dummyHasPrimaryContext(C10_UNUSED DeviceIndex device_index) { in dummyHasPrimaryContext()
184 bool hasPrimaryContext(DeviceIndex device_index) { in hasPrimaryContext()
H A DCUDAStream.h100 DeviceIndex device_index() const { in device_index() function
167 DeviceIndex device_index, in unpack3()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_cuda_ops.cpp17 void _device_synchronize(int64_t device_index) { in _device_synchronize()
41 c10::DeviceIndex device_index = device.has_value() in __anon4995d1b40202() local
53 c10::DeviceIndex device_index = in __anon4995d1b40302() local
64 c10::DeviceIndex device_index = device.has_value() in __anon4995d1b40402() local
76 c10::DeviceIndex device_index = in __anon4995d1b40502() local
171 c10::DeviceIndex device_index = device.has_value() in __anon4995d1b40e02() local
181 c10::DeviceIndex device_index = in __anon4995d1b40f02() local
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_cuda.cpp9 int32_t device_index, in aoti_torch_create_cuda_guard()
25 int32_t device_index) { in aoti_torch_cuda_guard_set_index()
33 int32_t device_index, in aoti_torch_create_cuda_stream_guard()
50 aoti_torch_get_current_cuda_stream(int32_t device_index, void** ret_stream) { in aoti_torch_get_current_cuda_stream()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Ddevice_spec.py112 device_index=None): argument
283 def device_index(self): member in DeviceSpecV2
370 def _components_to_string(job, replica, task, device_type, device_index): argument
441 def device_index(self, device_index): argument
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/
H A Dutils_cuda.h27 AOTICudaGuard(int32_t device_index) : guard_(nullptr, delete_cuda_guard) { in AOTICudaGuard()
34 void set_index(int32_t device_index) { in set_index()
45 AOTICudaStreamGuard(cudaStream_t stream, int32_t device_index) in AOTICudaStreamGuard()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAGeneratorImpl.cpp49 const Generator& getDefaultCUDAGenerator(DeviceIndex device_index) { in getDefaultCUDAGenerator()
67 Generator createCUDAGenerator(DeviceIndex device_index) { in createCUDAGenerator()
242 CUDAGeneratorImpl::CUDAGeneratorImpl(DeviceIndex device_index) in CUDAGeneratorImpl()
251 DeviceIndex device_index, in CUDAGeneratorImpl()
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp100 auto device_index = THPUtils_unpackDeviceIndex(arg); in THCPModule_exchangeDevice() local
115 auto device_index = THPUtils_unpackDeviceIndex(arg); in THCPModule_maybeExchangeDevice() local
189 PyObject* device_index) { in THCPModule_getCurrentStream_wrap()
210 PyObject* device_index) { in THCPModule_getCurrentStream_raw()
222 PyObject* device_index) { in THCPModule_getDefaultStream_wrap()
247 int64_t device_index = 0; in THCPModule_setStream_wrap() local
515 auto device_index = THPUtils_unpackDeviceIndex(arg); in THCPModule_hasPrimaryContext() local
538 auto device_index = THPUtils_unpackDeviceIndex(device_o); in THCPModule_setMemoryFraction() local
566 const auto device_index = THPUtils_unpackDeviceIndex(arg); in THCPModule_memoryStats() local
626 const auto device_index = THPUtils_unpackDeviceIndex(arg); in THCPModule_resetAccumulatedMemoryStats() local
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/mtia/
H A DModule.cpp53 m.def("_mtia_getCurrentStream", [](c10::DeviceIndex device_index) { in initModule()
64 m.def("_mtia_getDefaultStream", [](c10::DeviceIndex device_index) { in initModule()
78 m.def("_mtia_memoryStats", [](c10::DeviceIndex device_index) { in initModule()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSpectralOps.cpp92 CuFFTParamsLRUCache &cufft_get_plan_cache(DeviceIndex device_index) { in cufft_get_plan_cache()
111 int64_t cufft_get_plan_cache_max_size_impl(DeviceIndex device_index) { in cufft_get_plan_cache_max_size_impl()
119 void cufft_set_plan_cache_max_size_impl(DeviceIndex device_index, int64_t max_size) { in cufft_set_plan_cache_max_size_impl()
127 int64_t cufft_get_plan_cache_size_impl(DeviceIndex device_index) { in cufft_get_plan_cache_size_impl()
135 void cufft_clear_plan_cache_impl(DeviceIndex device_index) { in cufft_clear_plan_cache_impl()
/aosp_15_r20/external/pytorch/c10/core/
H A DStream.h26 DeviceIndex device_index; member
107 DeviceIndex device_index() const noexcept { in device_index() function
154 DeviceIndex device_index, in unpack3()
H A DTensorOptions.h217 device_index(c10::DeviceIndex device_index) const noexcept { in device_index() function
294 c10::DeviceIndex device_index() const noexcept { in device_index() function
601 inline TensorOptions device_index(c10::DeviceIndex device_index) { in device_index() argument
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DVirtualGuardImpl.h63 const DeviceIndex device_index, in record()
73 void destroyEvent(void* event, const DeviceIndex device_index) in destroyEvent()
90 double elapsedTime(void* event1, void* event2, const DeviceIndex device_index) in elapsedTime()
/aosp_15_r20/system/core/fs_mgr/liblp/include/liblp/
H A Dbuilder.h75 LinearExtent(uint64_t num_sectors, uint32_t device_index, uint64_t physical_sector) in LinearExtent()
86 uint32_t device_index() const { return device_index_; } in device_index() function
170 uint32_t device_index; member
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DCUDAHooks.cpp60 bool _hasPrimaryContext(DeviceIndex device_index) { in _hasPrimaryContext()
427 void CUDAHooks::cuFFTSetPlanCacheMaxSize(DeviceIndex device_index, int64_t max_size) const { in cuFFTSetPlanCacheMaxSize()
444 bool CUDAHooks::isGPUArch(DeviceIndex device_index, const std::vector<std::string>& archs) const { in isGPUArch()
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUStream.h74 DeviceIndex device_index() const { in device_index() function
128 DeviceIndex device_index, in unpack3()
/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPStreamMasqueradingAsCUDA.h51 DeviceIndex device_index() const { return stream_.device_index(); } in device_index() function
78 DeviceIndex device_index, in unpack3()
/aosp_15_r20/external/webrtc/modules/desktop_capture/win/
H A Dscreen_capture_utils.cc39 for (int device_index = 0;; ++device_index) { in GetScreenList() local
62 bool GetHmonitorFromDeviceIndex(const DesktopCapturer::SourceId device_index, in GetHmonitorFromDeviceIndex()
/aosp_15_r20/external/pytorch/torch/csrc/jit/cuda/
H A Dcuda.h18 c10::DeviceIndex device_index = variable
44 int64_t device_index() const { in device_index() function
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_lib.cc289 for (int device_index = 0; device_index < devices.size(); ++device_index) { in ParallelDevice() local
353 for (int device_index = 0; device_index < underlying_devices_.size(); in StartExecute() local
422 for (int device_index = 0; device_index < underlying_devices_.size(); in Join() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A Dcollective_ops_test.py644 def all_reduce(device, device_index): argument
650 def f(device_index): argument
682 def all_reduce(device, device_index): argument
688 def f(device_index): argument
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp557 PrivateGeneratorImpl(c10::DeviceIndex device_index) { in PrivateGeneratorImpl()
565 at::Generator make_generator_privateuse1(c10::DeviceIndex device_index) { in make_generator_privateuse1()
577 void set_custom_device_index(c10::DeviceIndex device_index) { in set_custom_device_index()
630 const at::Generator& default_generator(c10::DeviceIndex device_index) { in default_generator()
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_device.cpp185 PrivateGeneratorImpl(c10::DeviceIndex device_index) { in PrivateGeneratorImpl()
193 at::Generator make_generator_privateuse1(c10::DeviceIndex device_index) { in make_generator_privateuse1()

1234