Home
last modified time | relevance | path

Searched refs:MS_CTX_MAX_DEVICE_MEMORY (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_allocator.cc34 limited_device_memory_ = context_ptr->get_param<float>(MS_CTX_MAX_DEVICE_MEMORY); in Init()
51 auto max_device_memory = context_ptr->get_param<float>(MS_CTX_MAX_DEVICE_MEMORY); in CheckMaxDeviceMemory()
/third_party/mindspore/mindspore/ccsrc/pybind_api/utils/
Dms_context_py.cc86 .value("max_device_memory", MsCtxParam::MS_CTX_MAX_DEVICE_MEMORY) in __anon7eb2a90d0202()
/third_party/mindspore/mindspore/core/utils/
Dms_context.cc87 set_param<float>(MS_CTX_MAX_DEVICE_MEMORY, kDefaultMaxDeviceMemory); in MsContext()
Dms_context.h113 MS_CTX_MAX_DEVICE_MEMORY = MS_CTX_TYPE_FLOAT_BEGIN, enumerator