Searched refs:switch_pred (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | cond_builder.cc | 35 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/ |
| D | lower_if_op.cc | 151 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/ |
| D | keras_imagenet_main.pbtxt | 28665 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 …]
|