Searched refs:host_compute_builder (Results 1 – 1 of 1) sorted by relevance
316 NodeDefBuilder host_compute_builder(host_compute_node_name(original_oc_name), in BuildXlaHostComputeNodeDef() local321 host_compute_builder.Attr(kXlaOriginalOutsideCompilationNodeName, in BuildXlaHostComputeNodeDef()322 host_compute_builder.node_name()); in BuildXlaHostComputeNodeDef()326 host_compute_builder.Attr(attr.first, attr.second); in BuildXlaHostComputeNodeDef()333 host_compute_builder.Attr("tpu_core", core); in BuildXlaHostComputeNodeDef()354 host_compute_builder.Attr(kXlaTokenInputNodesAttrName, xla_token_input_nodes); in BuildXlaHostComputeNodeDef()372 host_compute_builder.Input(inputs); in BuildXlaHostComputeNodeDef()375 TF_RETURN_IF_ERROR(host_compute_builder.Finalize(&new_def)); in BuildXlaHostComputeNodeDef()