Searched refs:if_builder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 382 NodeDefBuilder if_builder("", "If"); in LazyConjunctionNodes() local
383 if_builder.Input(GetOutputNode(first_function, 0), 0, DT_BOOL); in LazyConjunctionNodes()
390 if_builder.Attr("Tin", in_arg_types); in LazyConjunctionNodes()
392 if_builder.Attr("Tcond", DT_BOOL); in LazyConjunctionNodes()
393 if_builder.Attr("Tout", DataTypeVector{DT_BOOL}); in LazyConjunctionNodes()
394 if_builder.Attr("_lower_using_switch_merge", true); in LazyConjunctionNodes()
398 if_builder.Attr("then_branch", then_branch); in LazyConjunctionNodes()
405 if_builder.Attr("else_branch", else_branch); in LazyConjunctionNodes()
406 if_builder.Input(inputs); in LazyConjunctionNodes()
410 TF_CHECK_OK(if_builder.Finalize(if_node)); in LazyConjunctionNodes()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 919 NodeDefBuilder if_builder(absl::StrCat("oc_if_", if_node_name), "If"); in BuildHostGraphForIfNode() local
920 if_builder.Attr("Tcond", DT_BOOL); in BuildHostGraphForIfNode()
921 if_builder.Attr("Tin", std::vector<DataType>{DT_STRING}); in BuildHostGraphForIfNode()
922 if_builder.Attr("Tout", std::vector<DataType>{}); in BuildHostGraphForIfNode()
928 if_builder.Attr("then_branch", host_then_branch); in BuildHostGraphForIfNode()
929 if_builder.Attr("else_branch", host_else_branch); in BuildHostGraphForIfNode()
930 if_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForIfNode()
931 if_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in BuildHostGraphForIfNode()
932 if_builder.Attr(outside_compilation_attr_name, outside_compilation_name); in BuildHostGraphForIfNode()
933 if_builder.Input(recv_pred_node->name(), 0, DT_BOOL); in BuildHostGraphForIfNode()
[all …]
|