Home
last modified time | relevance | path

Searched refs:switch_pred (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dcond_builder.cc35 Node* switch_pred; in CondBuilder() local
39 .Build(graph_, &switch_pred)); in CondBuilder()
40 graph_->AddEdge(pred(), 0, switch_pred, 0); in CondBuilder()
41 graph_->AddEdge(pred(), 0, switch_pred, 1); in CondBuilder()
50 graph_->AddEdge(switch_pred, kElseBranch, switch_f_, 0); in CondBuilder()
51 graph_->AddEdge(switch_pred, kThenBranch, switch_t_, 0); in CondBuilder()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op.cc151 Node* switch_pred; in CreatePivotNodes() local
158 graph_, &switch_pred)); in CreatePivotNodes()
159 control_predecessor_ = switch_pred; in CreatePivotNodes()
163 .Input(switch_pred, kElseBranch) in CreatePivotNodes()
169 .Input(switch_pred, kThenBranch) in CreatePivotNodes()
/external/tensorflow/tensorflow/compiler/jit/tests/
Dkeras_imagenet_main.pbtxt28665 name: "bn_conv1/cond_1/switch_pred/_163"
30143 name: "ConstantFoldingCtrl/bn_conv1/cond_1/switch_pred/_163_0"
30145 input: "bn_conv1/cond_1/switch_pred/_163"
30155 name: "ConstantFoldingCtrl/bn_conv1/cond_1/switch_pred/_163_1"
30157 input: "bn_conv1/cond_1/switch_pred/_163:1"
30924 input: "^ConstantFoldingCtrl/bn_conv1/cond_1/switch_pred/_163_0"
30950 input: "^ConstantFoldingCtrl/bn_conv1/cond_1/switch_pred/_163_0"
30976 input: "^ConstantFoldingCtrl/bn_conv1/cond_1/switch_pred/_163_0"
31002 input: "^ConstantFoldingCtrl/bn_conv1/cond_1/switch_pred/_163_0"
31028 input: "^ConstantFoldingCtrl/bn_conv1/cond_1/switch_pred/_163_0"
[all …]