Home
last modified time | relevance | path

Searched defs:CollectiveParams (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcollective.h111 struct CollectiveParams { struct
112 CollGroupParams group;
113 CollInstanceParams instance;
114 CollTaskParams task;
116 string name = ""; // node name used only for log or error messages
117 int default_rank = -1; // index of this op within device_names
118 bool is_source = false; // broadcast only
119 int source_rank = -1; // broadcast only
121 std::vector<int> subdiv_rank;
122 std::unique_ptr<OpKernel> merge_op; // reduction only
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster.h45 Status InitializeInstanceBeforeGroupDiscovery(CollectiveParams*) override { in InitializeInstanceBeforeGroupDiscovery() argument
Dring_alg.h45 Status InitializeInstanceBeforeGroupDiscovery(CollectiveParams*) override { in InitializeInstanceBeforeGroupDiscovery() argument