Home
last modified time | relevance | path

Searched refs:kControlSlot (Results 1 – 25 of 40) 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()
117 string control = TensorIdToString({to_node_name, Graph::kControlSlot}); in SwapControlledFanoutInputs()
153 auto from_control_fanouts = fanouts->find({from_node, Graph::kControlSlot}); in SwapFanoutInputs()
157 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.cc258 EXPECT_EQ(graph.HasFanin(*d_node, {"a", Graph::kControlSlot}), true); in TEST_F()
261 EXPECT_EQ(graph.HasFanin(*d_node, {"b", Graph::kControlSlot}), false); in TEST_F()
263 EXPECT_EQ(graph.HasFanin(*d_node, {"c", Graph::kControlSlot}), false); in TEST_F()
290 auto d_output_control = graph.GetRegularFanin({d_node, Graph::kControlSlot}); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc35 const int Graph::kControlSlot = -1; member in tensorflow::Graph
465 if (source == source_node() || dest == sink_node() || x == kControlSlot || in AddEdge()
466 y == kControlSlot) { in AddEdge()
467 DCHECK_EQ(x, kControlSlot) << source->DebugString(); in AddEdge()
468 DCHECK_EQ(y, kControlSlot) << dest->DebugString(); in AddEdge()
504 del->src_output_ = kControlSlot - 1; in RemoveEdge()
505 del->dst_input_ = kControlSlot - 1; in RemoveEdge()
536 return AddEdge(source, kControlSlot, dest, kControlSlot); in AddControlEdge()
613 if (src_slot == Graph::kControlSlot) { in AddInput()
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_constructor_test.cc133 return HasEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot); in HasControlEdge()
1283 const int kControlSlot = Graph::kControlSlot; in TEST_F() local
1284 opts.input_map[TensorId("W2", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1285 opts.input_map[TensorId("W3", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1321 opts.input_map[TensorId("W1", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1348 opts.input_map[TensorId("W2", Graph::kControlSlot)] = TensorId("W1", 0); in TEST_F()
1360 opts.input_map[TensorId("W2", 0)] = TensorId("W1", Graph::kControlSlot); in TEST_F()
1401 const int kControlSlot = Graph::kControlSlot; in TEST_F() local
1402 opts.input_map[TensorId("W2", kControlSlot)] = TensorId("DNE", kControlSlot); in TEST_F()
1459 const int kControlSlot = Graph::kControlSlot; in TEST_F() local
[all …]
Dgraph_partition.cc161 if (src_slot == Graph::kControlSlot) { in AddInput()
896 AddInput(dummy, src_name, Graph::kControlSlot); in AddControlEdges()
910 AddInput(ndef, dummy->name(), Graph::kControlSlot); in AddControlEdges()
1083 AddInput(dst_def, recv_node_name, Graph::kControlSlot); in Partition()
1111 AddInput(dummy, src->name(), Graph::kControlSlot); in Partition()
1134 AddInput(real_recv, send->name(), Graph::kControlSlot); in Partition()
1140 Graph::kControlSlot); in Partition()
1162 AddInput(dst_def, recv->name(), Graph::kControlSlot); in Partition()
1183 Graph::kControlSlot); in Partition()
Dgraph_constructor.cc421 if ((src.second == Graph::kControlSlot) != in ValidateInputMapAndControlDependencies()
422 (dst.second == Graph::kControlSlot)) { in ValidateInputMapAndControlDependencies()
726 if (new_input.second == Graph::kControlSlot) { in RemapNodeDefInputs()
771 string input = TensorId(control_dep, Graph::kControlSlot).ToString(); in AddControlDependencies()
1038 } else if (inputs[i].index == Graph::kControlSlot) { in Convert()
1071 if (e.src_index == Graph::kControlSlot) { in AddBackEdges()
1123 id.second != Graph::kControlSlot) { in PopulateReturnTensors()
Dsubgraph_test.cc104 return HasEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot); in HasControlEdge()
Dquantize_training.cc112 if (edge->src_output() != Graph::kControlSlot && edge->dst_input() == 0) { in FindType()
122 if (edge->src_output() != Graph::kControlSlot) { in FindType()
628 if (edge->src_output() == Graph::kControlSlot) { in DoQuantizeTraining()
Dgraph_test.cc126 e->src_output() == Graph::kControlSlot && in ControlEdgeExistsInGraphOrNodeDef()
127 e->dst_input() == Graph::kControlSlot) { in ControlEdgeExistsInGraphOrNodeDef()
461 EXPECT_EQ(edge->src_output(), Graph::kControlSlot); in TEST_F()
463 EXPECT_EQ(edge->dst_input(), Graph::kControlSlot); in TEST_F()
Dsubgraph.cc95 } else if (e->src_output() == Graph::kControlSlot && in FeedInputs()
111 CHECK_EQ(Graph::kControlSlot, e->src_output()); in FeedInputs()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h71 outside_port(Graph::kControlSlot), in EngineConnection()
74 inside_port(Graph::kControlSlot), in EngineConnection()
76 port_number(Graph::kControlSlot) {} in EngineConnection()
78 bool is_control_edge() const { return port_number == Graph::kControlSlot; } in is_control_edge()
Dconvert_graph.cc282 int port = Graph::kControlSlot - 1; in GetEngineInfo()
314 int port = Graph::kControlSlot - 1; in GetEngineInfo()
426 int port = Graph::kControlSlot; in CreateTRTNode()
430 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
617 QCHECK_EQ(Graph::kControlSlot, port); in CreateTRTNode()
/external/tensorflow/tensorflow/core/grappler/utils/
Dtopological_sort.cc41 GraphView::OutputPort(dep.from, Graph::kControlSlot), in MakeEphemeralEdges()
42 GraphView::InputPort(dep.to, Graph::kControlSlot)); in MakeEphemeralEdges()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc192 bool is_control = (out_port == Graph::kControlSlot); in AddEdge()
193 is_control |= (in_port == Graph::kControlSlot); in AddEdge()
201 AddEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot); in AddControlEdge()
367 graph->AddEdge(e->src(), e->src_output(), src, Graph::kControlSlot); in ContractEdge()
386 graph->AddEdge(src, Graph::kControlSlot, e->dst(), e->dst_input()); in ContractEdge()
/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/core/grappler/optimizers/data/
Dvectorization_utils.cc518 conversion_map_.insert({{arg_node, Graph::kControlSlot}, in AddArgTensorMappings()
519 {input_node, Graph::kControlSlot, stacked}}); in AddArgTensorMappings()
599 conversion_map_.insert({{tensor.first, Graph::kControlSlot}, in AddUnstackedTensorMappingsHelper()
600 WrappedTensor(node, Graph::kControlSlot, false)}); in AddUnstackedTensorMappingsHelper()
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis.cc814 SetPredicate(n, Graph::kControlSlot, in HandleSwitch()
939 SetPredicate(n, {0, 1, Graph::kControlSlot}, input_data_pred, in HandleMerge()
950 SetPredicate(n, {0, 1, Graph::kControlSlot}, input_data_pred, in HandleMerge()
983 SetPredicate(n, {0, 1, Graph::kControlSlot}, and_rec, should_revisit); in HandleMerge()
1001 SetPredicate(n, {0, Graph::kControlSlot}, in HandleRecv()
1016 SetPredicate(n, Graph::kControlSlot, pred, should_revisit); in HandleGeneric()
1027 SetPredicate(n, Graph::kControlSlot, predicate_factory_.MakeTrue(), in HandleNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc40 if (input.port_id == Graph::kControlSlot) { in IsTrivialIdentity()
49 if (output.port_id == Graph::kControlSlot) { in IsTrivialIdentity()
106 } else if (input.port_id == Graph::kControlSlot) { in CanRemoveNode()
Dfunction_optimizer.cc356 DCHECK(from.index() != Graph::kControlSlot) in AddTensorMapping()
358 DCHECK(to.index() != Graph::kControlSlot) in AddTensorMapping()
1602 ctx->graph_view().GetInputPort(func_node.name(), Graph::kControlSlot); in InlineIndirectFunctionCall()
1604 ctx->graph_view().GetOutputPort(func_node.name(), Graph::kControlSlot); in InlineIndirectFunctionCall()
1630 if (tensor_id.index() == Graph::kControlSlot) break; in InlineIndirectFunctionCall()
1948 if (input_tensor.index() == Graph::kControlSlot) break; in RestoreGraphInvariants()
1979 if (input_tensor.index() != Graph::kControlSlot) { in RestoreGraphInvariants()
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc207 graph->AddEdge(debug_node, Graph::kControlSlot, edge->dst(), in InsertNodes()
208 Graph::kControlSlot); in InsertNodes()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc829 if (arg.src_output == Graph::kControlSlot) { in AddInputEdges()
1197 dst_port == Graph::kControlSlot in RemoveRedundantMerge()
1198 ? Graph::kControlSlot in RemoveRedundantMerge()
1250 if (switch_branch == Graph::kControlSlot) { in RemoveRedundantSwitch()
1269 switch_branch == Graph::kControlSlot ? Graph::kControlSlot : val_port, in RemoveRedundantSwitch()

12