tvm_ffi::device

Function with_stream

Source 
pub fn with_stream<T>(
    device: &DLDevice,
    stream: TVMFFIStreamHandle,
    f: impl FnOnce() -> Result<T>,
) -> Result<T>
Expand description

call f with the stream set to the given stream