Home
last modified time | relevance | path

Searched defs:replicate_nodes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc333 Graph* graph, std::vector<Node*>* replicate_nodes, in FindTaggedNodes()
520 const Graph* graph, const std::vector<Node*>& replicate_nodes, in FindHostComputeKeyPlaceholderNodes()
4962 std::vector<Node*> replicate_nodes; in InternalRun() local
H A Dencapsulate_tpu_computations_pass.cc2499 std::vector<Node*> replicate_nodes; in BuildTPUReplicateOps() local