| /external/v8/src/compiler/ |
| raw-machine-assembler.cc | 102 Node* switch_node = AddNode(common()->Switch(succ_count), index); local
108 graph()->NewNode(common()->IfValue(case_value), switch_node);
114 Node* default_node = graph()->NewNode(common()->IfDefault(), switch_node);
118 schedule()->AddSwitch(CurrentBlock(), switch_node, succ_blocks, succ_count); local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session_test.cc | 614 Node* switch_node = test::graph::Switch(&g, bool_const, bool_const); local
615 Node* fourth_identity = test::graph::Identity(&g, switch_node, 1);
626 Status s = session->PRunSetup({switch_node->name() + ":1"},
636 // Feed switch_node:1 and fetch fourth_identity.
637 s = session->PRun(handle, {{switch_node->name() + ":1", bool_value}},
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| quantize_training.cc | 431 Node* switch_node; local
435 .Finalize(graph, &switch_node));
436 NodeBuilder::NodeOut output_false = NodeBuilder::NodeOut(switch_node, 0);
437 NodeBuilder::NodeOut output_true = NodeBuilder::NodeOut(switch_node, 1);
|
| graph_partition.cc | 453 Node* switch_node = nullptr; member in struct:tensorflow::__anon39621::ControlLoop
507 Node* switch_node = AddControlSwitch(merge, loop_cond, device_name,
511 AddControlNext({switch_node, 1}, device_name, bopts.WithName(next_name));
517 AddControlFlowInfo(switch_node, src, cf_info);
526 loop->switch_node = switch_node;
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| layout_optimizer_test.cc | 1012 auto switch_node = node_map.GetNode("switch"); local
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.cc | 57 Node* switch_node = nullptr; member in struct:tensorflow::__anon38720::Arg
264 node_map[arg.switch_node->id()] = arg_node;
269 squash_src_outputs[arg.switch_node->id()] = true;
410 if (arg.switch_node != nullptr) {
414 arg.switch_node = edge->dst();
417 if (arg.switch_node == nullptr) {
432 for (const Edge* edge : arg.switch_node->out_edges()) {
448 arg.switch_node->name());
454 arg.switch_node->name(),
[all...] |
