Searched refs:GraphCollectiveOrder (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | build_graph_options.cc | 40 case GraphCollectiveOrder::kNone: in DebugString()
43 case GraphCollectiveOrder::kEdges: in DebugString()
46 case GraphCollectiveOrder::kAttrs: in DebugString()
|
| D | build_graph_options.h | 41 GraphCollectiveOrder collective_order = GraphCollectiveOrder::kNone;
|
| D | graph_execution_state.cc | 948 if (options.collective_order != GraphCollectiveOrder::kNone) { in BuildGraph()
|
| D | direct_session.cc | 1303 options.collective_order = GraphCollectiveOrder::kEdges; in CreateExecutors()
1305 options.collective_order = GraphCollectiveOrder::kAttrs; in CreateExecutors()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order.h | 22 enum class GraphCollectiveOrder { kNone, kEdges, kAttrs }; enum
32 Status OrderCollectives(Graph* graph, GraphCollectiveOrder order_type);
|
| D | collective_order.cc | 148 Graph* graph, GraphCollectiveOrder order_type, in InsertControlDependencies()
151 if (order_type == GraphCollectiveOrder::kEdges) { in InsertControlDependencies()
158 } else if (order_type == GraphCollectiveOrder::kAttrs) { in InsertControlDependencies()
184 Status OrderCollectives(Graph* graph, GraphCollectiveOrder order_type) { in OrderCollectives()
|
| D | collective_order_test.cc | 143 TF_EXPECT_OK(OrderCollectives(graph.get(), GraphCollectiveOrder::kEdges)); in TEST()
150 TF_EXPECT_OK(OrderCollectives(graph.get(), GraphCollectiveOrder::kAttrs)); in TEST()
192 TF_EXPECT_OK(OrderCollectives(graph.get(), GraphCollectiveOrder::kEdges)); in TEST()
230 TF_EXPECT_OK(OrderCollectives(graph.get(), GraphCollectiveOrder::kAttrs)); in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 1162 opts->collective_order = GraphCollectiveOrder::kEdges; in BuildBuildGraphOptions()
1164 opts->collective_order = GraphCollectiveOrder::kAttrs; in BuildBuildGraphOptions()
|