Searched refs:kPrimDType (Results 1 – 3 of 3) sorted by relevance
34 DType() : PrimitiveC(prim::kPrimDType->name()) { InitIOName({"x"}, {"output"}); } in DType()
50 REGISTER_PRIMITIVE_EVAL_IMPL(DType, prim::kPrimDType, DTypeInfer, DTypeInferValue, false);
676 inline const PrimitivePtr kPrimDType = std::make_shared<Primitive>("DType"); variable