Home
last modified time | relevance | path

Searched defs:cpu_device_context (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Dembedding_cache_prefetch_actor.h362 explicit Sender(device::DeviceContext *cpu_device_context) in Sender()
410 explicit Receiver(device::DeviceContext *cpu_device_context) in Receiver()
Dembedding_cache_prefetch_actor.cc60 … int32_t param_key, device::DeviceContext *cpu_device_context) { in CreateSenderReceiverPair()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dgraph_compiler.cc723 … auto cpu_device_context = device::DeviceContextManager::GetInstance().GetOrCreateDeviceContext( in CompileGraphImpl() local