Searched refs:prim2 (Results 1 – 10 of 10) sorted by relevance
91 auto prim2 = new schema::ActivationT; in BuildGraph() local92 prim2->activation_type = activation_type; in BuildGraph()93 next_node->primitive->value.value = prim2; in BuildGraph()
93 auto prim2 = new schema::BatchNormT; in BuildCaffeGraph() local94 bn_node->primitive->value.value = prim2; in BuildCaffeGraph()174 auto prim2 = new schema::FusedBatchNormT; in BuildTFGraph() local175 bn_node->primitive->value.value = prim2; in BuildTFGraph()
92 auto prim2 = new schema::BiasAddT; in BuildGraph() local93 biasadd_node->primitive->value.value = prim2; in BuildGraph()
110 auto prim2 = new schema::ScaleFusionT; in BuildGraph() local111 scale_node->primitive->value.value = prim2; in BuildGraph()
128 PrimitivePtr prim2 = MatMul_2_node->input(0)->cast<ValueNodePtr>()->value()->cast<PrimitivePtr>(); in Create_two_nodes() local131 prim2->set_attr("transpose_a", transpose_a); in Create_two_nodes()132 prim2->set_attr("transpose_b", transpose_b); in Create_two_nodes()
175 PrimitivePtr prim2 = node2->input(0)->cast<ValueNodePtr>()->value()->cast<PrimitivePtr>(); in Make_Manager() local176 prim2->AddAttr("transpose_a", transpose_a); in Make_Manager()177 prim2->AddAttr("transpose_b", transpose_b); in Make_Manager()178 prim2->AddAttr("instance_name", MakeValue("matmul2")); in Make_Manager()179 prim2->AddAttr("strategy", var2); in Make_Manager()
547 boost::tiny_xml::element_ptr prim2(new boost::tiny_xml::element()); in process_node() local548 prim2->name = "primary"; in process_node()549 … prim2->elements.push_front(boost::tiny_xml::element_ptr(new boost::tiny_xml::element())); in process_node()550 prim2->elements.front()->content = i->term; in process_node()551 p2->elements.push_front(prim2); in process_node()
220 bool IsPrimitiveEquals(const PrimitivePtr &prim1, const PrimitivePtr &prim2) { in IsPrimitiveEquals() argument221 if (prim1 == nullptr || prim2 == nullptr) { in IsPrimitiveEquals()224 return (prim1 == prim2) || (prim1->Hash() == prim2->Hash() && prim1->name() == prim2->name()); in IsPrimitiveEquals()
583 bool IsPrimitiveEquals(const PrimitivePtr &prim1, const PrimitivePtr &prim2);
246 auto prim2 = GetCNodePrimitive(cnode2); in CompFunc() local248 MS_EXCEPTION_IF_NULL(prim2); in CompFunc()250 auto rank_tag2 = prim2->GetAttr(SRC_RANK); in CompFunc()255 rank_tag2 = prim2->GetAttr(DEST_RANK); in CompFunc()264 auto sr_tag2 = prim2->GetAttr(SR_TAG); in CompFunc()