Home
last modified time | relevance | path

Searched defs:switch_node (Results 1 – 16 of 16) sorted by relevance

/external/python/pycparser/pycparser/
Dast_transforms.py13 def fix_switch_cases(switch_node): argument
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc89 auto switch_node = in TEST() local
143 auto switch_node = in TEST() local
Dmark_for_compilation_pass.cc1078 const Node* switch_node = (*it)->src(); in IsIdentityDrivingConstsInLoop() local
Dmark_for_compilation_pass_test.cc1682 ops::Switch switch_node(root.WithOpName("switch"), value, loop_cond); in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.h43 Node* switch_node = nullptr; member
Dfunctionalize_cond.cc124 Status GetSwitchPredicate(const Node& switch_node, OutputTensor* pred) { in GetSwitchPredicate()
136 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue()
510 Node* switch_node; in AddSwitchNodeAlongEdge() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc247 const Node* switch_node = nullptr; in TEST() local
368 const Node* switch_node = nullptr; in TEST() local
Dlower_while_op.cc330 Node* switch_node; in CreateSwitchNodes() local
Dquantize_training.cc435 Node* switch_node; in MakeInitializedEMAVariable() local
Ddirect_session_test.cc1646 Node* switch_node = test::graph::Switch(&g, bool_const, bool_const); in TEST() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc168 NodeDef* switch_node = nullptr; in HandleConst() local
598 const NodeDef& switch_node, const NodeMap& node_map, in CheckForDeadFanout()
Dlayout_optimizer_test.cc1117 auto switch_node = node_map.GetNode("switch"); in TEST_F() local
Dgeneric_layout_optimizer_transposer_test.cc1808 auto* switch_node = context.graph_view->GetNode("switch"); in TEST_F() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc285 auto switch_node = e_cond->dst(); in FindLoopExitNodes() local
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc496 Node* switch_node = nullptr; member
550 Node* switch_node = AddControlSwitch(merge, loop_cond, device_name, in AddControlLoop() local
Dgraph_partition_test.cc650 Switch switch_node(scope, merge.output, loop_cond); in TEST() local