Searched defs:CollectiveParams (Results 1 – 3 of 3) sorted by relevance
111 struct CollectiveParams { struct112 CollGroupParams group;113 CollInstanceParams instance;114 CollTaskParams task;116 string name = ""; // node name used only for log or error messages117 int default_rank = -1; // index of this op within device_names118 bool is_source = false; // broadcast only119 int source_rank = -1; // broadcast only121 std::vector<int> subdiv_rank;122 std::unique_ptr<OpKernel> merge_op; // reduction only[all …]
45 Status InitializeInstanceBeforeGroupDiscovery(CollectiveParams*) override { in InitializeInstanceBeforeGroupDiscovery() argument