Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc57 const Graph& graph, const std::unordered_set<Node*>& loop_nodes, // NOLINT in IsExecuteNodeOrIdentityToExecuteNode()
94 const Graph& graph, const std::unordered_set<Node*>& loop_nodes, // NOLINT in ResourceOnlyUsedForTPUExecuteInLoop()
115 const std::unordered_set<Node*>& loop_nodes, // NOLINT in ExtractExecuteNodeInfo()
H A Dhost_training_loop_optimization_util.h57 std::unordered_set<Node*> loop_nodes; // NOLINT member