Searched defs:set_device_memory_capacity (Results 1 – 2 of 2) sorted by relevance
125 void CostModelContext::set_device_memory_capacity(double dm_capacity) { in set_device_memory_capacity() function in mindspore::parallel::CostModelContext
46 def set_device_memory_capacity(self, dev_mem_cap): member in _CostModelContext