Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dconfigure_tpu_embedding_rewrite_pass.cc239 const std::string& embedding_attr_string = GetNodeAttrString( in Run() local
241 if (embedding_attr_string.empty()) { in Run()
248 tpu_embedding_config.ParseFromString(embedding_attr_string); in Run()
H A Ddistributed_tpu_configuration_rewrite_pass.cc248 const std::string& embedding_attr_string = GetNodeAttrString( in Run() local
251 if (!embedding_attr_string.empty()) { in Run()