Home
last modified time | relevance | path

Searched defs:compute_stream (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.h71 void *compute_stream() const override { return stream_; } in compute_stream() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.h52 void *compute_stream() const override { return stream_; } in compute_stream() function
Dgpu_stream_assign.cc39 CudaDeviceStream compute_stream = GPUDeviceManager::GetInstance().default_stream(); in AssignGpuStream() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.h115 virtual void *compute_stream() const { return nullptr; } in compute_stream() function
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc742 auto compute_stream = kernel_runtime->compute_stream(); in CreateBucket() local
Dascend_session.cc516 auto compute_stream = runtime_instance->compute_stream(); in LoadInputData() local
1947 auto compute_stream = kernel_runtime->compute_stream(); in CreateBucket() local