Searched defs:Participant (Results 1 – 1 of 1) sorted by relevance
57 struct Participant { struct58 Participant(se::StreamExecutor* executor, se::Stream* tensor_stream, in Participant() argument76 se::StreamExecutor* const executor = nullptr;84 se::Stream* const tensor_stream;89 EventMgr* const event_mgr;91 const int gpu_device_id;95 const Tensor* input;99 Tensor* output;103 const int global_rank;108 DoneCallback done_callback;[all …]