Searched refs:kPrimEmbeddingLookup (Results 1 – 6 of 6) sorted by relevance
39 if (AnfAlgo::GetCNodeName(cnode) == prim::kPrimEmbeddingLookup->name() || in Process()
44 Register(prim::kPrimEmbeddingLookup->name(), {2, 3, 4, 5}); in ConstInputToAttrInfoRegistry()
114 {prim::kPrimEmbeddingLookup, {InferImplEmbeddingLookup, nullptr, true}}, in GetPrimitiveToEvalImplMap()
448 IsPrimitiveCNode(cnodes[i], prim::kPrimEmbeddingLookup)) { in FindSparseGatherV2WithCache()
187 inline const PrimitivePtr kPrimEmbeddingLookup = std::make_shared<Primitive>("EmbeddingLookup"); variable
514 if (device_target != kCPUDevice && op_run_info->op_name == prim::kPrimEmbeddingLookup->name()) { in ConstructInputTensor()