| /external/tensorflow/tensorflow/core/graph/ |
| tensor_id.h | 38 if (second == Graph::kControlSlot) return strings::StrCat("^", first);
|
| tensor_id.cc | 51 id.second = Graph::kControlSlot;
|
| graph.cc | 34 const int Graph::kControlSlot = -1;
382 if (source == source_node() || dest == sink_node() || x == kControlSlot ||
383 y == kControlSlot) {
384 DCHECK_EQ(x, kControlSlot) << source->DebugString();
385 DCHECK_EQ(y, kControlSlot) << dest->DebugString();
421 del->src_output_ = kControlSlot - 1;
422 del->dst_input_ = kControlSlot - 1;
453 return AddEdge(source, kControlSlot, dest, kControlSlot);
510 if (src_slot == Graph::kControlSlot) {
[all...] |
| graph_constructor_test.cc | 133 return HasEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot);
1278 const int kControlSlot = Graph::kControlSlot;
1279 opts.input_map[TensorId("W2", kControlSlot)] = TensorId("W1", kControlSlot);
1280 opts.input_map[TensorId("W3", kControlSlot)] = TensorId("W1", kControlSlot);
[all...] |
| graph_partition.cc | 158 if (src_slot == Graph::kControlSlot) {
[all...] |
| mkl_tfconversion_pass_test.cc | 65 } else if (index == Graph::kControlSlot) {
|
| subgraph.cc | 117 } else if (e->src_output() == Graph::kControlSlot &&
133 CHECK_EQ(Graph::kControlSlot, e->src_output());
|
| subgraph_test.cc | 104 return HasEdge(src, Graph::kControlSlot, dst, Graph::kControlSlot);
|
| graph_constructor.cc | 393 if ((src.second == Graph::kControlSlot) !=
394 (dst.second == Graph::kControlSlot)) {
692 if (new_input.second == Graph::kControlSlot) {
737 string input = TensorId(control_dep, Graph::kControlSlot).ToString();
[all...] |
| quantize_training.cc | 112 if (edge->src_output() != Graph::kControlSlot && edge->dst_input() == 0) {
122 if (edge->src_output() != Graph::kControlSlot) {
622 if (edge->src_output() == Graph::kControlSlot) {
|
| graph_test.cc | 124 e->src_output() == Graph::kControlSlot &&
125 e->dst_input() == Graph::kControlSlot) {
459 EXPECT_EQ(edge->src_output(), Graph::kControlSlot);
461 EXPECT_EQ(edge->dst_input(), Graph::kControlSlot);
|
| optimizer_cse_test.cc | 61 } else if (index == Graph::kControlSlot) {
|
| graph.h | 302 // carried by this edge. The special value kControlSlot is used
307 // carried by this edge. The special value kControlSlot is used
411 static const int kControlSlot;
785 // Note that if either src_output_ or dst_input_ is kControlSlot,
787 return src_output_ == Graph::kControlSlot;
|
| mkl_layout_pass_test.cc | 77 } else if (index == Graph::kControlSlot) {
[all...] |
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_rewrite.cc | 93 if (out_node.index == Graph::kControlSlot) {
149 if (out_node.index == Graph::kControlSlot) {
190 if (out_node.index == Graph::kControlSlot) {
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| segment.cc | 87 tensorflow::Graph::kControlSlot);
99 graph->AddEdge(src, tensorflow::Graph::kControlSlot, e->dst(),
|
| /external/tensorflow/tensorflow/core/debug/ |
| debug_graph_utils.cc | 207 graph->AddEdge(debug_node, Graph::kControlSlot, edge->dst(),
208 Graph::kControlSlot);
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fold_constants_lib.cc | 91 if (id.second != Graph::kControlSlot) {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.cc | 317 e->IsControlEdge() ? Graph::kControlSlot : 0);
536 if (dst_input == Graph::kControlSlot) {
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| utils.cc | 177 if (output_index == Graph::kControlSlot) {
|
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| executor.cc | [all...] |
