Searched refs:kGatherV2OpName (Results 1 – 3 of 3) sorted by relevance
498 …if (cnode_name == kEmbeddingLookupOpName || cnode_name == kGatherV2OpName || cnode_name == kSparse… in GetEmbeddingTableParamPtr()524 if (cnode_name != kGatherV2OpName && cnode_name != kSparseGatherV2OpName) { in CacheEmbeddingTableParamPtr()
241 constexpr auto kGatherV2OpName = "Gather"; variable
1583 if (kernel_name != kGatherV2OpName && kernel_name != kSparseGatherV2OpName) { in GetFirstPSEmbeddingCache()1657 if (kernel_name != kGatherV2OpName && kernel_name != kSparseGatherV2OpName) { in CheckIfSupportPSEmbeddingCache()