Home
last modified time | relevance | path

Searched defs:async_comms_stream (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dthunk.cc24 se::Stream* async_comms_stream) in ExecuteParams()
H A Dgpu_executable.cc326 StatusOr<StreamPool::Ptr> async_comms_stream = in ExecuteThunks() local
638 StatusOr<StreamPool::Ptr> async_comms_stream = in ExecuteJitRt() local
H A Dthunk.h113 se::Stream* async_comms_stream; member
H A Dnccl_all_reduce_thunk.cc286 se::Stream& async_comms_stream = *params.async_comms_stream; in RunNcclCollective() local