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