Home
last modified time | relevance | path

Searched refs:host_compute_key_placeholder_node (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.h566 FunctionLibraryRuntime* flr, Node* host_compute_key_placeholder_node,
H A Ddistributed_tpu_rewrite_pass.cc4713 FunctionLibraryRuntime* flr, Node* host_compute_key_placeholder_node, in RewriteTPUReplicateNode() argument
4858 if (host_compute_key_placeholder_node != nullptr) { in RewriteTPUReplicateNode()
4860 compile_node, host_compute_key_placeholder_node, graph)); in RewriteTPUReplicateNode()