Searched defs:set_device_id (Results 1 – 4 of 4) sorted by relevance
95 void set_device_id(uint32_t device_id) { device_id_ = device_id; } in set_device_id() function
104 void GpuBufferMgr::set_device_id(int device_id) { cur_dev_id_ = device_id; } in set_device_id() function in mindspore::device::GpuBufferMgr
217 def set_device_id(self, device_id): member in _Context
270 int set_device_id = static_cast<int>(device_info_->GetDeviceID()); in SetCudaDevice() local