Searched defs:SetThreadDevice (Results 1 – 4 of 4) sorted by relevance
94 void GpuDataQueueDynamic::SetThreadDevice() { SetThreadToDeviceId(device_id_); } in SetThreadDevice() function in mindspore::device::GpuDataQueueDynamic173 void GpuQueue::SetThreadDevice() { SetThreadToDeviceId(device_id_); } in SetThreadDevice() function in mindspore::device::GpuQueue265 void GpuDataQueue::SetThreadDevice() { SetThreadToDeviceId(device_id_); } in SetThreadDevice() function in mindspore::device::GpuDataQueue
56 virtual void SetThreadDevice() {} in SetThreadDevice() function
243 DataQueueStatus DataQueueMgr::SetThreadDevice(const std::string &channel_name) const { in SetThreadDevice() function in mindspore::device::DataQueueMgr
797 Status DataQueueOp::SetThreadDevice() { in SetThreadDevice() function in mindspore::dataset::DataQueueOp