Searched refs:recv_at_host_builder (Results 1 – 1 of 1) sorted by relevance
110 NodeDefBuilder recv_at_host_builder( in BuildRecvAtHostNode() local114 recv_at_host_builder.Attr("Toutputs", recv_at_host_dtypes); in BuildRecvAtHostNode()119 recv_at_host_builder.Attr("device_ordinal", device_ordinal_value); in BuildRecvAtHostNode()120 recv_at_host_builder.Attr( in BuildRecvAtHostNode()122 recv_at_host_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildRecvAtHostNode()123 recv_at_host_builder.Input(key_placeholder->name(), 0, DT_STRING); in BuildRecvAtHostNode()124 TF_RETURN_IF_ERROR(recv_at_host_builder.Finalize(&recv_at_host_def)); in BuildRecvAtHostNode()