Home
last modified time | relevance | path

Searched defs:device_context_key (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_device_context.h107 …explicit CPUDeviceContext(const DeviceContextKey &device_context_key) : DeviceInterface(device_con… in CPUDeviceContext()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_device_context.h178 …explicit GPUDeviceContext(const DeviceContextKey &device_context_key) : DeviceInterface(device_con… in GPUDeviceContext()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_device_context.h47 …explicit GeDeviceContext(const DeviceContextKey &device_context_key) : DeviceInterface(device_cont… in GeDeviceContext()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/
Dgraph_compiler_test.cc79 DeviceContextKey device_context_key{"CPU", 0}; in TEST_F() local
Dgraph_scheduler_test.cc150 DeviceContextKey device_context_key{device_name, device_id}; in TEST_F() local
222 DeviceContextKey device_context_key{device_name, device_id}; in TEST_F() local
Dany_type_kernel_actor_test.cc108 DeviceContextKey device_context_key{device_name, device_id}; in TEST_F() local
Dcontrol_node_parser_test.cc179 DeviceContextKey device_context_key{"CPU", 0}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/common/
Dbackend_common_test.cc133 runtime::test::DeviceContextKey device_context_key{kDefaultDeviceName, 0}; in Compile() local
Ddevice_common_test.h209 …explicit TestDeviceContext(const DeviceContextKey &device_context_key) : DeviceInterface(device_co… in TestDeviceContext()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/hardware/
Ddevice_context.h72 explicit DeviceContext(const DeviceContextKey &device_context_key) in DeviceContext()
93 const DeviceContextKey &device_context_key() const { return device_context_key_; } in device_context_key() function
Ddevice_context_manager.cc440 …ontext *DeviceContextManager::GetOrCreateDeviceContext(const DeviceContextKey &device_context_key)… in GetOrCreateDeviceContext()