Searched refs:send_pred_builder (Results 1 – 1 of 1) sorted by relevance
1327 NodeDefBuilder send_pred_builder(name, "XlaSendToHost"); in BuildSendIfPredNode() local1328 send_pred_builder.Attr("Tinput", DT_BOOL); in BuildSendIfPredNode()1329 send_pred_builder.Attr("key", absl::StrCat(host_transfer_key, "_dtoh_0")); in BuildSendIfPredNode()1330 send_pred_builder.Attr(kXlaTokenInputNodesAttrName, in BuildSendIfPredNode()1332 send_pred_builder.Attr(kXlaOriginalOutsideCompilationNodeName, name); in BuildSendIfPredNode()1333 send_pred_builder.Input(pred_node->name(), 0, DT_BOOL); in BuildSendIfPredNode()1335 TF_RETURN_IF_ERROR(send_pred_builder.Finalize(&send_pred_def)); in BuildSendIfPredNode()