Searched defs:CollectiveParams (Results 1 – 2 of 2) sorted by relevance
135 struct CollectiveParams : public core::RefCounted { struct136 CollGroupParams group;137 CollInstanceParams instance;138 CollTaskParams task;140 string name = ""; // node name used only for log or error messages141 int default_rank = -1; // index of this op within device_names142 bool is_source = false; // broadcast only143 int source_rank = -1; // broadcast only145 std::vector<int> subdiv_rank;146 OpKernel* merge_op = nullptr; // reduction only[all …]
365 Status InitializeCollectiveParams(CollectiveParams*) override { in InitializeCollectiveParams() argument