Home
last modified time | relevance | path

Searched refs:kControlSlot (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view_test.cc43 if (tensor_id.index() == Graph::kControlSlot) { in CompareNodeFanins()
44 port = Graph::kControlSlot; in CompareNodeFanins()
965 TestAddRegularFanin("foo_1", /*node_exists=*/true, {"b", Graph::kControlSlot}, in TEST()
971 TestAddRegularFanin("foo_3", /*node_exists=*/true, {"c", Graph::kControlSlot}, in TEST()
977 TestAddRegularFanin("foo_2", /*node_exists=*/true, {"d", Graph::kControlSlot}, in TEST()
983 TestAddRegularFanin("foo_4", /*node_exists=*/true, {"a", Graph::kControlSlot}, in TEST()
990 TestAddRegularFanin("foo_5", /*node_exists=*/true, {"a", Graph::kControlSlot}, in TEST()
996 TestAddRegularFanin("foo_6", /*node_exists=*/true, {"c", Graph::kControlSlot}, in TEST()
1002 TestAddRegularFanin("foo_2", /*node_exists=*/true, {"a", Graph::kControlSlot}, in TEST()
1029 {"bar_missing", Graph::kControlSlot}, in TEST()
[all …]
Dmutable_graph_view.cc44 return tensor_id.index() >= Graph::kControlSlot; in IsTensorIdPortValid()
48 return tensor_id.index() > Graph::kControlSlot; in IsTensorIdRegular()
52 return tensor_id.index() == Graph::kControlSlot; in IsTensorIdControlling()
56 return port.port_id == Graph::kControlSlot; in IsOutputPortControlling()
118 string control = TensorIdToString({to_node_name, Graph::kControlSlot}); in SwapControlledFanoutInputs()
154 auto from_control_fanouts = fanouts->find({from_node, Graph::kControlSlot}); in SwapFanoutInputs()
158 auto to_control_fanouts = fanouts->find({to_node, Graph::kControlSlot}); in SwapFanoutInputs()
365 fanouts()[output].emplace(node, Graph::kControlSlot); in AddAndDedupFanouts()
513 MutableGraphView::OutputPort control_port(node, Graph::kControlSlot); in UpdateNode()
571 auto control_fanouts = fanouts().find({node, Graph::kControlSlot}); in UpdateNodeName()
[all …]
Dgraph_view.cc78 const auto control_port = GraphView::InputPort(node, Graph::kControlSlot); in HasControlFanin()
83 const auto control_port = GraphView::OutputPort(node, Graph::kControlSlot); in HasControlFanout()
Dgraph_view_test.cc259 EXPECT_EQ(graph.HasFanin(*d_node, {"a", Graph::kControlSlot}), true); in TEST_F()
262 EXPECT_EQ(graph.HasFanin(*d_node, {"b", Graph::kControlSlot}), false); in TEST_F()
264 EXPECT_EQ(graph.HasFanin(*d_node, {"c", Graph::kControlSlot}), false); in TEST_F()
291 auto d_output_control = graph.GetRegularFanin({d_node, Graph::kControlSlot}); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dtensor_id.cc57 id.second = Graph::kControlSlot; in ParseTensorName()
66 return tensor_id.index() == Graph::kControlSlot; in IsTensorIdControl()
Dtensor_id.h48 if (second == Graph::kControlSlot) return strings::StrCat("^", first); in ToString()
80 if (second == Graph::kControlSlot) return strings::StrCat("^", first); in ToString()
Dgraph.cc39 const int Graph::kControlSlot = -1; member in tensorflow::Graph
522 if (source == source_node() || dest == sink_node() || x == kControlSlot || in AddEdge()
523 y == kControlSlot) { in AddEdge()
524 DCHECK_EQ(x, kControlSlot) << source->DebugString(); in AddEdge()
525 DCHECK_EQ(y, kControlSlot) << dest->DebugString(); in AddEdge()
590 return AddEdge(source, kControlSlot, dest, kControlSlot); in AddControlEdge()
667 if (src_slot == Graph::kControlSlot) { in AddInput()
Dgraph_partition.cc163 if (src_slot == Graph::kControlSlot) { in AddInput()
923 AddInput(dummy, src_name, Graph::kControlSlot); in AddControlEdges()
937 AddInput(ndef, dummy->name(), Graph::kControlSlot); in AddControlEdges()
1110 AddInput(dst_def, recv_node_name, Graph::kControlSlot); in Partition()
1138 AddInput(dummy, src->name(), Graph::kControlSlot); in Partition()
1161 AddInput(real_recv, send->name(), Graph::kControlSlot); in Partition()
1167 Graph::kControlSlot); in Partition()
1189 AddInput(dst_def, recv->name(), Graph::kControlSlot); in Partition()
1210 Graph::kControlSlot); in Partition()
Dsubgraph_test.cc104 return HasEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot); in HasControlEdge()
Dsubgraph.cc94 } else if (e->src_output() == Graph::kControlSlot && in FeedInputs()
110 CHECK_EQ(Graph::kControlSlot, e->src_output()); in FeedInputs()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.cc49 if (fanin.index() < Graph::kControlSlot || graph_view_ != fanin.graph_view_) { in HasFanin()
57 if (fanout.index() < Graph::kControlSlot || in HasFanout()
64 } else if (fanout.index() == Graph::kControlSlot) { in HasFanout()
65 return view->fanins_set_.contains({this->node(), Graph::kControlSlot}); in HasFanout()
148 Graph::kControlSlot); in CheckAndAddFaninsInternal()
150 Graph::kControlSlot); in CheckAndAddFaninsInternal()
152 Graph::kControlSlot); in CheckAndAddFaninsInternal()
184 if (fanin.index() < Graph::kControlSlot || graph_view_ != fanin.graph_view_) { in HasFanin()
192 if (fanout.index() < Graph::kControlSlot || in HasFanout()
199 } else if (fanout.index() == Graph::kControlSlot) { in HasFanout()
[all …]
Dtopological_sort.cc41 GraphView::OutputPort(dep.from, Graph::kControlSlot), in MakeEphemeralEdges()
42 GraphView::InputPort(dep.to, Graph::kControlSlot)); in MakeEphemeralEdges()
Dgraph_view_test.cc384 const auto& a_fanin_control = a_node->GetRegularFanin(Graph::kControlSlot); in TYPED_TEST()
410 Graph::kControlSlot)); in TYPED_TEST()
414 Graph::kControlSlot); in TYPED_TEST()
510 EXPECT_EQ(d_node->GetRegularFanout(Graph::kControlSlot), no_fanouts); in TYPED_TEST()
541 FaninViewType(&graph_view, b_node->node_index(), Graph::kControlSlot))); in TYPED_TEST()
543 FaninViewType(&graph_view, a_node->node_index(), Graph::kControlSlot))); in TYPED_TEST()
547 Graph::kControlSlot); in TYPED_TEST()
677 {&graph_view, c_node->node_index(), Graph::kControlSlot})); in TYPED_TEST()
680 {&graph_view, b_node->node_index(), Graph::kControlSlot})); in TYPED_TEST()
709 {&graph_view, c_node->node_index(), Graph::kControlSlot})); in TYPED_TEST()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor_test.cc142 return HasEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot); in HasControlEdge()
1287 const int kControlSlot = Graph::kControlSlot; in TEST_F() local
1288 opts.input_map[TensorId("W2", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1289 opts.input_map[TensorId("W3", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1325 opts.input_map[TensorId("W1", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1352 opts.input_map[TensorId("W2", Graph::kControlSlot)] = TensorId("W1", 0); in TEST_F()
1364 opts.input_map[TensorId("W2", 0)] = TensorId("W1", Graph::kControlSlot); in TEST_F()
1405 const int kControlSlot = Graph::kControlSlot; in TEST_F() local
1406 opts.input_map[TensorId("W2", kControlSlot)] = TensorId("DNE", kControlSlot); in TEST_F()
1463 const int kControlSlot = Graph::kControlSlot; in TEST_F() local
[all …]
Dgraph_constructor.cc370 return input.index == Graph::kControlSlot; in IsControlInput()
627 if ((src.second == Graph::kControlSlot) != in ValidateInputMapAndControlDependencies()
628 (dst.second == Graph::kControlSlot)) { in ValidateInputMapAndControlDependencies()
879 if (new_input.second == Graph::kControlSlot) { in RemapNodeDefInputs()
924 string input = TensorId(control_dep, Graph::kControlSlot).ToString(); in AddControlDependencies()
1272 } else if (inputs[i].index == Graph::kControlSlot) { in Convert()
1306 if (e.src_index == Graph::kControlSlot) { in AddBackEdges()
1359 id.second != Graph::kControlSlot) { in PopulateReturnTensors()
Dquantize_training.cc110 if (edge->src_output() != Graph::kControlSlot && edge->dst_input() == 0) { in FindType()
120 if (edge->src_output() != Graph::kControlSlot) { in FindType()
626 if (edge->src_output() == Graph::kControlSlot) { in DoQuantizeTraining()
Dmkl_tfconversion_pass_test.cc64 } else if (index == Graph::kControlSlot) { in EdgeId()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc95 if (out_node.index == Graph::kControlSlot) { in ReplaceReduce()
151 if (out_node.index == Graph::kControlSlot) { in ReplaceBroadcast()
192 if (out_node.index == Graph::kControlSlot) { in ReplaceBroadcast()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h65 outside_port(Graph::kControlSlot), in EngineConnection()
68 inside_port(Graph::kControlSlot), in EngineConnection()
70 port_number(Graph::kControlSlot) {} in EngineConnection()
72 bool is_control_edge() const { return port_number == Graph::kControlSlot; } in is_control_edge()
Dconvert_graph.cc197 int port = Graph::kControlSlot - 1; in GetEngineInfo()
231 int port = Graph::kControlSlot - 1; in GetEngineInfo()
355 int port = Graph::kControlSlot; in CreateTRTNode()
359 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
546 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc511 conversion_map_.insert({{arg_node, Graph::kControlSlot}, in AddArgTensorMappings()
512 {input_node, Graph::kControlSlot, stacked}}); in AddArgTensorMappings()
592 conversion_map_.insert({{tensor.first, Graph::kControlSlot}, in AddUnstackedTensorMappingsHelper()
593 WrappedTensor(node, Graph::kControlSlot, false)}); in AddUnstackedTensorMappingsHelper()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc199 bool is_control = (out_port == Graph::kControlSlot); in AddEdge()
200 is_control |= (in_port == Graph::kControlSlot); in AddEdge()
208 AddEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot); in AddControlEdge()
595 graph->AddEdge(e->src(), e->src_output(), src, Graph::kControlSlot); in ContractEdge()
614 graph->AddEdge(src, Graph::kControlSlot, e->dst(), e->dst_input()); in ContractEdge()
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis.cc977 SetPredicate(n, Graph::kControlSlot, in HandleSwitch()
1137 SetPredicate(n, {0, 1, Graph::kControlSlot}, input_data_pred, in HandleMerge()
1148 SetPredicate(n, {0, 1, Graph::kControlSlot}, input_data_pred, in HandleMerge()
1181 SetPredicate(n, {0, 1, Graph::kControlSlot}, and_rec, should_revisit); in HandleMerge()
1199 SetPredicate(n, {0, Graph::kControlSlot}, in HandleRecv()
1214 SetPredicate(n, Graph::kControlSlot, pred, should_revisit); in HandleGeneric()
1226 SetPredicate(n, Graph::kControlSlot, predicate_factory_.MakeTrue(), in HandleNode()
1530 predicate_map_.erase(TensorId(n->name(), Graph::kControlSlot)); in PopulateFrame()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc41 if (input.port_id == Graph::kControlSlot) { in IsTrivialIdentity()
50 if (output.port_id == Graph::kControlSlot) { in IsTrivialIdentity()
124 } else if (input.port_id == Graph::kControlSlot) { in CanRemoveNode()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc173 if (dep.dst_input == Graph::kControlSlot) { in AddSynchronizationNode()
203 if (dep.dst_input != Graph::kControlSlot) { in AddShutdownNode()

12