Searched defs:ForwardPassAllreduceWork (Results 1 – 1 of 1) sorted by relevance
476 struct ForwardPassAllreduceWork { struct in c10d::Reducer477 c10::intrusive_ptr<c10d::Work> workHandle;478 at::Tensor resultTensor;481 bool useStaticWorldSize;