Home
last modified time | relevance | path

Searched refs:DeviceStore (Results 1 – 22 of 22) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/storage/
H A Dpinned_memory_storage.h59 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
68 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in Free()
H A Dgpu_device_storage.h61 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
75 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in Free()
H A Dpooled_storage_manager.h91 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in DirectFreeNoLock()
147 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
286 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in DirectFreeNoLock()
330 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/storage/
H A Dpinned_memory_storage.h59 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
68 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in Free()
H A Dgpu_device_storage.h61 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
75 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in Free()
H A Dpooled_storage_manager.h91 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in DirectFreeNoLock()
147 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
286 mxnet::common::cuda::DeviceStore device_store(handle.ctx.real_dev_id(), true); in DirectFreeNoLock()
330 mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true); in Alloc()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/engine/
H A Dstream_manager.h78 mxnet::common::cuda::DeviceStore device_store(ctx.dev_id); in GetRunContext()
120 mxnet::common::cuda::DeviceStore device_store(ctx.dev_id); in GetIORunContext()
H A Dthreaded_engine_pooled.cc136 mxnet::common::cuda::DeviceStore device_store(-1, false); in DoExecute()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/engine/
H A Dstream_manager.h78 mxnet::common::cuda::DeviceStore device_store(ctx.dev_id); in GetRunContext()
120 mxnet::common::cuda::DeviceStore device_store(ctx.dev_id); in GetIORunContext()
H A Dthreaded_engine_pooled.cc136 mxnet::common::cuda::DeviceStore device_store(-1, false); in DoExecute()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/common/
H A Dcuda_utils.h370 class DeviceStore {
373 explicit DeviceStore(int requested_device = -1, bool restore = true) :
384 ~DeviceStore() {
H A Drtc.cc80 mxnet::common::cuda::DeviceStore device_store; in GetFunction()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/common/
H A Dcuda_utils.h370 class DeviceStore {
373 explicit DeviceStore(int requested_device = -1, bool restore = true) :
384 ~DeviceStore() {
H A Drtc.cc80 mxnet::common::cuda::DeviceStore device_store; in GetFunction()
/dports/net-im/py-matrix-synapse/synapse-1.50.2/synapse/storage/databases/main/
H A D__init__.py38 from .devices import DeviceStore
116 DeviceStore,
H A Ddevices.py1186 class DeviceStore(DeviceWorkerStore, DeviceBackgroundUpdateStore): class
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/kvstore/
H A Dkvstore_nccl.h431 mxnet::common::cuda::DeviceStore device_store; in CommSync()
483 mxnet::common::cuda::DeviceStore device_store; in InitNCCL()
H A Dcomm_tree.h338 mxnet::common::cuda::DeviceStore device_store(gpus[i]); in EnableP2P()
H A Dcomm.h726 mxnet::common::cuda::DeviceStore device_store(gpus[i]); in EnableP2P()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/kvstore/
H A Dkvstore_nccl.h431 mxnet::common::cuda::DeviceStore device_store; in CommSync()
483 mxnet::common::cuda::DeviceStore device_store; in InitNCCL()
H A Dcomm_tree.h338 mxnet::common::cuda::DeviceStore device_store(gpus[i]); in EnableP2P()
H A Dcomm.h726 mxnet::common::cuda::DeviceStore device_store(gpus[i]); in EnableP2P()