Searched defs:CreateStream (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ | ||
| D | hccl_task.cc | 154 void HcclTask::CreateStream(int64_t stream_num, int64_t master_stream_id) { in CreateStream() function in mindspore::ge::model_runner::HcclTask |
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ | ||
| D | gpu_device_manager.cc | 72 bool GPUDeviceManager::CreateStream(CudaDeviceStream *stream) { in CreateStream() function in mindspore::device::gpu::GPUDeviceManager |
| D | cuda_driver.cc | 147 bool CudaDriver::CreateStream(CudaDeviceStream *stream) { in CreateStream() function in mindspore::device::gpu::CudaDriver |