| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 269 Node* switch_node; in CreateSwitchNodes() local
279 .Finalize(graph_, &switch_node)); in CreateSwitchNodes()
280 switch_nodes_.emplace_back(switch_node); in CreateSwitchNodes()
286 for (Node* switch_node : switch_nodes_) { in CreateBodyFuncCallNode()
287 body_call_builder_.Input(NodeOut(switch_node, 1)); in CreateBodyFuncCallNode()
315 for (Node* switch_node : switch_nodes_) { in CreateExitNodes()
319 .Input(NodeOut(switch_node, 0)) in CreateExitNodes()
|
| D | direct_session_test.cc | 1222 Node* switch_node = test::graph::Switch(&g, bool_const, bool_const); in TEST() local
1223 Node* fourth_identity = test::graph::Identity(&g, switch_node, 1); in TEST()
1234 Status s = session->PRunSetup({switch_node->name() + ":1"}, in TEST()
1245 s = session->PRun(handle, {{switch_node->name() + ":1", bool_value}}, in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 57 Node* switch_node = nullptr; member
221 node_map[arg.switch_node->id()] = arg_node; in BuildLoopBody()
226 squash_src_outputs[arg.switch_node->id()] = true; in BuildLoopBody()
420 if (arg.switch_node != nullptr) { in FunctionalizeLoop()
424 arg.switch_node = edge->dst(); in FunctionalizeLoop()
427 if (arg.switch_node == nullptr) { in FunctionalizeLoop()
442 for (const Edge* edge : arg.switch_node->out_edges()) { in FunctionalizeLoop()
459 FormatNodeForError(*arg.switch_node)); in FunctionalizeLoop()
465 FormatNodeForError(*arg.switch_node), in FunctionalizeLoop()
|
| D | functionalize_cond.cc | 120 Status GetSwitchPredicate(const Node& switch_node, OutputTensor* pred) { in GetSwitchPredicate() argument
122 TF_RETURN_IF_ERROR(switch_node.input_edge(1, &pred_edge)); in GetSwitchPredicate()
132 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue() argument
134 TF_RETURN_IF_ERROR(switch_node.input_edge(0, &val_edge)); in GetSwitchValue()
412 for (Node* switch_node : switches_) { in BuildArgumentNodes()
414 TF_RETURN_IF_ERROR(switch_node->input_edge(0, &e)); in BuildArgumentNodes()
420 cond_arg_nodes_.at(input_index.at(key)).switches.push_back(switch_node); in BuildArgumentNodes()
494 Node* switch_node; in AddSwitchNodeAlongEdge() local
502 .Finalize(graph, &switch_node)); in AddSwitchNodeAlongEdge()
503 state_map_->ResetCondId(switch_node, state_map_->LookupCondId(src)); in AddSwitchNodeAlongEdge()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 87 auto switch_node = in TEST() local
90 switch_node.output_false); in TEST()
92 switch_node.output_true); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 170 NodeDef* switch_node = nullptr; in HandleConst() local
173 switch_node = node; in HandleConst()
177 if (!switch_node) { in HandleConst()
181 string switch_output = StrCat(switch_node->name(), ":1"); in HandleConst()
596 const NodeDef& switch_node, const NodeMap& node_map, in CheckForDeadFanout() argument
601 GraphView::InputPort switch_loopcond_port(&switch_node, 1); in CheckForDeadFanout()
613 GraphView::InputPort switch_input_port(&switch_node, 0); in CheckForDeadFanout()
|
| D | layout_optimizer_test.cc | 1097 auto switch_node = node_map.GetNode("switch"); in TEST_F() local
1098 EXPECT_EQ(switch_node->input(0), "Conv2D"); in TEST_F()
1099 EXPECT_EQ(switch_node->input(1), "ctrl"); in TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | quantize_training.cc | 437 Node* switch_node; in MakeInitializedEMAVariable() local
441 .Finalize(graph, &switch_node)); in MakeInitializedEMAVariable()
442 NodeBuilder::NodeOut output_false = NodeBuilder::NodeOut(switch_node, 0); in MakeInitializedEMAVariable()
443 NodeBuilder::NodeOut output_true = NodeBuilder::NodeOut(switch_node, 1); in MakeInitializedEMAVariable()
|
| D | graph_partition.cc | 469 Node* switch_node = nullptr; member
523 Node* switch_node = AddControlSwitch(merge, loop_cond, device_name, in AddControlLoop() local
527 AddControlNext({switch_node, 1}, device_name, bopts.WithName(next_name)); in AddControlLoop()
533 AddControlFlowInfo(switch_node, src, cf_info); in AddControlLoop()
542 loop->switch_node = switch_node; in AddControlLoop()
|
| D | graph_partition_test.cc | 650 Switch switch_node(scope, merge.output, loop_cond); in TEST() local
651 Identity identity(scope, switch_node.output_true); in TEST()
654 switch_node.output_false); in TEST()
663 enter, merge.output, cv, loop_cond, switch_node.output_false, in TEST()
|
| /external/v8/src/compiler/ |
| D | raw-machine-assembler.cc | 122 Node* switch_node = AddNode(common()->Switch(succ_count), index); in Switch() local
128 graph()->NewNode(common()->IfValue(case_value), switch_node); in Switch()
134 Node* default_node = graph()->NewNode(common()->IfDefault(), switch_node); in Switch()
138 schedule()->AddSwitch(CurrentBlock(), switch_node, succ_blocks, succ_count); in Switch()
|