Searched refs:kPrimDiag (Results 1 – 3 of 3) sorted by relevance
77 auto diag_prim = std::make_shared<Primitive>(prim::kPrimDiag->name()); in DefinePattern()95 …std::vector<AnfNodePtr> new_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimDiag->name(… in Process()
32 Diag() : PrimitiveC(prim::kPrimDiag->name()) { InitIOName({"input_x"}, {"output"}); } in Diag()
256 inline const PrimitivePtr kPrimDiag = std::make_shared<Primitive>(kDiag); variable