Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dshape_inference_test.cc89 auto switch_node = in TEST() local
143 auto switch_node = in TEST() local
H A Dpartially_decluster_pass_test.cc530 ops::Switch switch_node(root.WithOpName("switch"), value, loop_cond); in TEST() local
H A Dmark_for_compilation_pass.cc1171 const Node* switch_node = (*it)->src(); in IsIdentityDrivingConstsInLoop() local
H A Dmark_for_compilation_pass_test.cc1721 ops::Switch switch_node(root.WithOpName("switch"), value, loop_cond); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc286 auto switch_node = e_cond->dst(); in FindLoopExitNodes() local
362 for (auto switch_node : loop_cond_node.out_nodes()) { in AddNoOpAfterLastIteration() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_while_op_test.cc247 const Node* switch_node = nullptr; in TEST() local
368 const Node* switch_node = nullptr; in TEST() local
H A Dlower_while_op.cc333 Node* switch_node; in CreateSwitchNodes() local
H A Dquantize_training.cc435 Node* switch_node; in MakeInitializedEMAVariable() local
H A Ddirect_session_test.cc1673 Node* switch_node = test::graph::Switch(&g, bool_const, bool_const); in TEST() local
/aosp_15_r20/external/python/pycparser/pycparser/
Dast_transforms.py13 def fix_switch_cases(switch_node): argument
/aosp_15_r20/external/ot-br-posix/tests/scripts/expect/
H A D_common.exp97 proc switch_node {id} { procedure
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_cond.cc122 Status GetSwitchPredicate(const Node& switch_node, OutputTensor* pred) { in GetSwitchPredicate()
134 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue()
508 Node* switch_node; in AddSwitchNodeAlongEdge() local
H A Dfunctionalize_control_flow_util.h43 Node* switch_node = nullptr; member
/aosp_15_r20/external/openthread/tests/scripts/expect/
H A D_common.exp149 proc switch_node {id} { procedure
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dloop_optimizer.cc168 NodeDef* switch_node = nullptr; in HandleConst() local
598 const NodeDef& switch_node, const NodeMap& node_map, in CheckForDeadFanout()
H A Dgeneric_layout_optimizer_transposer_test.cc1808 auto* switch_node = context.graph_view->GetNode("switch"); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dgraph_partition.cc495 Node* switch_node = nullptr; member
549 Node* switch_node = AddControlSwitch(merge, loop_cond, device_name, in AddControlLoop() local
H A Dgraph_partition_test.cc651 Switch switch_node(scope, merge.output, loop_cond); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state.cc508 const NodeDef* switch_node = nullptr; in FindLoopCondFromExitNode() local