Searched refs:send_from_host_builder (Results 1 – 1 of 1) sorted by relevance
212 NodeDefBuilder send_from_host_builder( in BuildSendFromHostNode() local216 send_from_host_builder.Attr("Tinputs", send_from_host_dtypes); in BuildSendFromHostNode()221 send_from_host_builder.Attr("device_ordinal", device_ordinal_value); in BuildSendFromHostNode()222 send_from_host_builder.Attr( in BuildSendFromHostNode()224 send_from_host_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildSendFromHostNode()239 send_from_host_builder.Input(inputs); in BuildSendFromHostNode()240 send_from_host_builder.Input(key_placeholder->name(), 0, DT_STRING); in BuildSendFromHostNode()241 TF_RETURN_IF_ERROR(send_from_host_builder.Finalize(&send_from_host_def)); in BuildSendFromHostNode()