Searched refs:kPrimDiagPart (Results 1 – 3 of 3) sorted by relevance
24 auto diag_apart_prim = std::make_shared<Primitive>(prim::kPrimDiagPart->name()); in DefinePattern()46 …nfNodePtr> new_node_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimDiagPart->name()))}; in Process()
32 DiagPart() : PrimitiveC(prim::kPrimDiagPart->name()) { InitIOName({"input_x"}, {"output"}); } in DiagPart()
257 inline const PrimitivePtr kPrimDiagPart = std::make_shared<Primitive>(kDiagPart); variable