set_stream

Module.set_stream(stream: Stream)

Sets a stream to be used for model inference.

Parameters:

stream (Stream) -- The stream to be set.