Home
last modified time | relevance | path

Searched defs:SetCudaStream (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_allocator.h45 void SetCudaStream(cudaStream_t stream) { stream_ = stream; } in SetCudaStream() function
Dtensorrt_runtime.h60 …void SetCudaStream(cudaStream_t stream, cublasHandle_t cublas_handle, cublasLtHandle_t cublaslt_ha… in SetCudaStream() function