Searched defs:embedding_attr (Results 1 – 1 of 1) sorted by relevance
53 auto embedding_attr = op->getAttrOfType<StringAttr>(kTPUEmbeddingAttr); in ExtractEmbeddingAttribute() local99 const auto& embedding_attr = it.getKey(); in UpdateEmbeddingEnqueueOpInput() local