Searched defs:async_comms_stream (Results 1 – 4 of 4) sorted by relevance
24 se::Stream* async_comms_stream) in ExecuteParams()
113 se::Stream* async_comms_stream; member
326 StatusOr<StreamPool::Ptr> async_comms_stream = in ExecuteThunks() local638 StatusOr<StreamPool::Ptr> async_comms_stream = in ExecuteJitRt() local
286 se::Stream& async_comms_stream = *params.async_comms_stream; in RunNcclCollective() local