Searched defs:compute_stream (Results 1 – 6 of 6) sorted by relevance
71 void *compute_stream() const override { return stream_; } in compute_stream() function
52 void *compute_stream() const override { return stream_; } in compute_stream() function
39 CudaDeviceStream compute_stream = GPUDeviceManager::GetInstance().default_stream(); in AssignGpuStream() local
115 virtual void *compute_stream() const { return nullptr; } in compute_stream() function
742 auto compute_stream = kernel_runtime->compute_stream(); in CreateBucket() local
516 auto compute_stream = runtime_instance->compute_stream(); in LoadInputData() local1947 auto compute_stream = kernel_runtime->compute_stream(); in CreateBucket() local