Searched refs:recv_pred_builder (Results 1 – 1 of 1) sorted by relevance
1409 NodeDefBuilder recv_pred_builder( in BuildHostGraphForIfNode() local1411 recv_pred_builder.Attr("Toutputs", std::vector<DataType>{DT_BOOL}); in BuildHostGraphForIfNode()1412 recv_pred_builder.Attr("key", host_transfer_key); in BuildHostGraphForIfNode()1413 recv_pred_builder.Attr("device_ordinal", device_ordinal_value); in BuildHostGraphForIfNode()1414 recv_pred_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in BuildHostGraphForIfNode()1415 recv_pred_builder.Attr(outside_compilation_attr_name, in BuildHostGraphForIfNode()1417 recv_pred_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForIfNode()1418 recv_pred_builder.Input(key_placeholder->name(), 0, DT_STRING); in BuildHostGraphForIfNode()1420 TF_RETURN_IF_ERROR(recv_pred_builder.Finalize(&recv_pred_def)); in BuildHostGraphForIfNode()1582 NodeDefBuilder recv_pred_builder( in RewriteHostWhileLoopCond() local[all …]