Searched defs:CollectiveParams (Results 1 – 1 of 1) sorted by relevance
144 struct CollectiveParams : public core::RefCounted { struct145 CollGroupParams group;146 CollInstanceParams instance;148 string name = ""; // node name used only for log or error messages149 int default_rank = -1; // index of this op within device_names150 bool is_source = false; // broadcast only151 int source_rank = -1; // broadcast only153 std::vector<int> subdiv_rank;154 OpKernel* merge_op = nullptr; // reduction only155 OpKernel* final_op = nullptr; // reduction only[all …]