Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h553 FunctionLibraryRuntime* flr, Node* host_compute_key_placeholder_node,
Ddistributed_tpu_rewrite_pass.cc4212 FunctionLibraryRuntime* flr, Node* host_compute_key_placeholder_node, in RewriteTPUReplicateNode() argument
4353 if (host_compute_key_placeholder_node != nullptr) { in RewriteTPUReplicateNode()
4355 compile_node, host_compute_key_placeholder_node, graph)); in RewriteTPUReplicateNode()