Searched refs:if_builder (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 383 NodeDefBuilder if_builder("", "If"); in LazyConjunctionNodes() local
384 if_builder.Input(GetOutputNode(first_function, 0), 0, DT_BOOL); in LazyConjunctionNodes()
391 if_builder.Attr("Tin", in_arg_types); in LazyConjunctionNodes()
393 if_builder.Attr("Tcond", DT_BOOL); in LazyConjunctionNodes()
394 if_builder.Attr("Tout", DataTypeVector{DT_BOOL}); in LazyConjunctionNodes()
395 if_builder.Attr("_lower_using_switch_merge", true); in LazyConjunctionNodes()
399 if_builder.Attr("then_branch", then_branch); in LazyConjunctionNodes()
406 if_builder.Attr("else_branch", else_branch); in LazyConjunctionNodes()
407 if_builder.Input(inputs); in LazyConjunctionNodes()
411 TF_CHECK_OK(if_builder.Finalize(if_node)); in LazyConjunctionNodes()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 370 void createRankSpecializedBroadcastAndOp(OpBuilder &if_builder, ChloOpTy op, in createRankSpecializedBroadcastAndOp()
377 createBroadcastToKnownRank(if_builder, op, lhs, targeted_rank); in createRankSpecializedBroadcastAndOp()
379 createBroadcastToKnownRank(if_builder, op, rhs, targeted_rank); in createRankSpecializedBroadcastAndOp()
390 Value reshaped_lhs = if_builder.create<mhlo::DynamicReshapeOp>( in createRankSpecializedBroadcastAndOp()
392 Value reshaped_rhs = if_builder.create<mhlo::DynamicReshapeOp>( in createRankSpecializedBroadcastAndOp()
400 Value result = if_builder.create<ChloOpTy>( in createRankSpecializedBroadcastAndOp()
403 Value reshaped_result = if_builder.create<tensor::CastOp>( in createRankSpecializedBroadcastAndOp()
405 if_builder.create<scf::YieldOp>(loc, reshaped_result); in createRankSpecializedBroadcastAndOp()
434 OpBuilder if_builder = if_op.getThenBodyBuilder(rewriter.getListener()); in HandleBroadcastAndOp() local
435 createRankSpecializedBroadcastAndOp(if_builder, op, lhs, rhs, 1); in HandleBroadcastAndOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1434 NodeDefBuilder if_builder(absl::StrCat("oc_if_", if_node_name), "If"); in BuildHostGraphForIfNode() local
1435 if_builder.Attr("Tcond", DT_BOOL); in BuildHostGraphForIfNode()
1436 if_builder.Attr("Tin", std::vector<DataType>{DT_STRING}); in BuildHostGraphForIfNode()
1437 if_builder.Attr("Tout", std::vector<DataType>{}); in BuildHostGraphForIfNode()
1443 if_builder.Attr("then_branch", host_then_branch); in BuildHostGraphForIfNode()
1444 if_builder.Attr("else_branch", host_else_branch); in BuildHostGraphForIfNode()
1445 if_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForIfNode()
1446 if_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in BuildHostGraphForIfNode()
1447 if_builder.Attr(outside_compilation_attr_name, outside_compilation_name); in BuildHostGraphForIfNode()
1448 if_builder.Input(recv_pred_node->name(), 0, DT_BOOL); in BuildHostGraphForIfNode()
[all …]
|