Searched refs:set_device_memory_capacity (Results 1 – 4 of 4) sorted by relevance
42 def set_device_memory_capacity(self, dev_mem_cap): member in _CostModelContext54 self._context_handle.set_device_memory_capacity(dev_mem_cap)504 "device_memory_capacity": cost_model_context().set_device_memory_capacity,
64 void set_device_memory_capacity(double);
116 void CostModelContext::set_device_memory_capacity(double dm_capacity) { in set_device_memory_capacity() function in mindspore::parallel::CostModelContext
208 .def("set_device_memory_capacity", &CostModelContext::set_device_memory_capacity, in PYBIND11_MODULE()