Searched refs:kPrimEmbed (Results 1 – 6 of 6) sorted by relevance
376 FuncGraphPtr graph_embed = MakeFuncGraph(prim::kPrimEmbed, 1); in TEST_F()396 FuncGraphPtr graph_embed = MakeFuncGraph(prim::kPrimEmbed, 1); in TEST_F()426 FuncGraphPtr graph_embed = MakeFuncGraph(prim::kPrimEmbed, 1); in TEST_F()
125 embed_node = tape_->NewCNode({NewValueNode(prim::kPrimEmbed), fv_node}); in BackPropagateFv()381 auto node = tape_->NewCNode({NewValueNode(prim::kPrimEmbed), fv_adjoint->second->k()}); in AttachFvDoutToTape()408 auto node = tape_->NewCNode({NewValueNode(prim::kPrimEmbed), fv_adjoint.second->k()}); in AttachIndirectFvDoutToTape()
346 ASSERT_EQ(prim->name(), kPrimEmbed->name()); in TEST_F()
292 AnfNodePtr reverted_node = root->NewCNode({NewValueNode(prim::kPrimEmbed), param}); in RevertSymbolicKeyInstance()
688 inline const PrimitivePtr kPrimEmbed = std::make_shared<Primitive>("embed"); variable
1474 constructor[prim::kPrimEmbed] = std::make_shared<EmbedEvaluator>(); in InitPrimEvaluatorConstructors()