Searched defs:set_current_stream (Results 1 – 2 of 2) sorted by relevance
94 void set_current_stream(size_t stream_id) { current_stream_id_ = stream_id; } in set_current_stream() function
161 void GPUDeviceManager::set_current_stream(size_t stream_id) { current_stream_id_ = stream_id; } in set_current_stream() function in mindspore::device::gpu::GPUDeviceManager