tensorflow/core/tpu/graph_rewrite/combine_tpu_embedding_load_retrieve_pass.cc
// Returns names of the load and retrieve node of all optimizers supported
// by TPUEmbedding.
Returns names of the load and retrieve node of all optimizers supported
by TPUEmbedding.
Returns names of the load and retrieve node of all optimizers supported by TPUEmbedding .
absl::flat_hash_set<std::string> GetLoadRetrieveNodeNames() {
  std::vector<std::string> load_names =
      GetPerTableLoadOptimizationParametersOps();
  std::vector<std::string> retrieve_names =
      GetPerTableRetrieveOptimizationParametersOps();
  absl::flat_hash_set<std::string> nodes(load_names.begin(), load_names.end());
  nodes.insert(retrieve_names.begin(), retrieve_names.end());
  return nodes;
}
absl :: flat_hash_set < std :: string > GetLoadRetrieveNodeNames ( ) { std :: vector < std :: string > load_names = GetPerTableLoadOptimizationParametersOps ( ) ; std :: vector < std :: string > retrieve_names = GetPerTableRetrieveOptimizationParametersOps ( ) ; absl :: flat_hash_set < std :: string > nodes ( load_names . begin ( ) , load_names . end ( ) ) ; nodes . insert ( retrieve_names . begin ( ) , retrieve_names . end ( ) ) ; return nodes ; }
Returns names of the load and retrieve node of all optimizers supported
by TPUEmbedding.
Returns names of the load and retrieve node of all optimizers supported by TPUEmbedding .