Searched defs:NcclExecuteParams (Results 1 – 2 of 2) sorted by relevance
84 struct NcclExecuteParams { struct88 se::Stream* stream;89 RunId run_id;90 const DeviceAssignment* device_assn; // never null91 const std::vector<GlobalDeviceId>* gpu_global_device_ids; // may be null92 const NcclUniqueIdCallback* nccl_unique_id_callback; // may be null
53 NcclExecuteParams::NcclExecuteParams( in NcclExecuteParams() function in xla::gpu::NcclExecuteParams