Searched refs:pred_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1325 const string& name, const string& host_transfer_key, Node* pred_node, in BuildSendIfPredNode() argument
1333 send_pred_builder.Input(pred_node->name(), 0, DT_BOOL); in BuildSendIfPredNode()
1339 g->AddEdge(pred_node, 0, send_pred_node, 0); in BuildSendIfPredNode()
1948 Node* pred_node; in ExtractOutsideCompilationForIfNode() local
1949 TF_RETURN_IF_ERROR(n->input_node(0, &pred_node)); in ExtractOutsideCompilationForIfNode()
1953 host_transfer_key, pred_node, g)); in ExtractOutsideCompilationForIfNode()
|