Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc87 auto switch_node = in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op.cc269 Node* switch_node; in CreateSwitchNodes() local
Ddirect_session_test.cc1222 Node* switch_node = test::graph::Switch(&g, bool_const, bool_const); in TEST() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc170 NodeDef* switch_node = nullptr; in HandleConst() local
596 const NodeDef& switch_node, const NodeMap& node_map, in CheckForDeadFanout()
Dlayout_optimizer_test.cc1097 auto switch_node = node_map.GetNode("switch"); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc120 Status GetSwitchPredicate(const Node& switch_node, OutputTensor* pred) { in GetSwitchPredicate()
132 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue()
494 Node* switch_node; in AddSwitchNodeAlongEdge() local
Dfunctionalize_while.cc57 Node* switch_node = nullptr; member
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc469 Node* switch_node = nullptr; member
523 Node* switch_node = AddControlSwitch(merge, loop_cond, device_name, in AddControlLoop() local
Dquantize_training.cc437 Node* switch_node; in MakeInitializedEMAVariable() local
Dgraph_partition_test.cc650 Switch switch_node(scope, merge.output, loop_cond); in TEST() local
/external/v8/src/compiler/
Draw-machine-assembler.cc122 Node* switch_node = AddNode(common()->Switch(succ_count), index); in Switch() local