Home
last modified time | relevance | path

Searched refs:InferType (Results 1 – 25 of 104) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/
Dop_node.h70 …virtual TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) { return inputs[0]->type;… in InferType() function
96 TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override;
106 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return inputs[2]->type… in InferType() function
117 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return inputs[1]->type… in InferType() function
126 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return TypeId::kNumber… in InferType() function
205 TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override;
225 TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override;
258 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return TypeId::kNumber… in InferType() function
273 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return TypeId::kNumber… in InferType() function
283 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return TypeId::kNumber… in InferType() function
[all …]
/third_party/typescript/tests/cases/compiler/
DpropTypeValidatorInference.ts9 export type InferPropsInner<V> = { [K in keyof V]-?: InferType<V[K]>; };
22 export type InferType<V> = V extends Validator<infer T> ? T : any; alias
33 export function oneOfType<T extends Validator<any>>(types: T[]): Requireable<NonNullable<InferType<…
/third_party/mindspore/mindspore/core/ops/
Dround.cc30 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
39 …return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), InferShape(inp… in RoundInfer()
Dsin.cc39 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
50 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in SinInfer()
Dmaximum.cc32 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
45 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in MaximumInfer()
Ddiv.cc35 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
48 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in DivInfer()
Dpow.cc31 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
44 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in PowInfer()
Dnot_equal.cc30 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
48 (void)InferType(primitive, input_args); in NotEqualInfer()
Dones_like.cc36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
50 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in OnesLikeInfer()
Dless.cc34 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
48 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in LessInfer()
Dassign_add.cc32 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
43 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in AssignAddInfer()
Dless_equal.cc35 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
48 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in LessEqualInfer()
Dlogical_and.cc35 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
49 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in LogicalAndInfer()
Dsquared_difference.cc33 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
47 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in SquaredDifferenceInfer()
Dhshrink.cc35 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
46 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in HShrinkInfer()
Dfloor.cc34 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
46 …return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), InferShape(inp… in FloorInfer()
Dlogical_or.cc36 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
50 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in LogicalOrInfer()
Dzeros_like.cc36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
50 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in ZerosLikeInfer()
Dhsigmoid.cc33 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
45 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in HSigmoidInfer()
Drsqrt.cc40 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
50 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in RsqrtInfer()
Drelu.cc41 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
53 auto type = InferType(primitive, input_args); in ReLUInfer()
Dmul.cc41 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
58 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in MulInfer()
Dsub.cc41 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
57 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in SubInfer()
Dminimum.cc36 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
53 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in MinimumInfer()
/third_party/mindspore/mindspore/core/ops/grad/
Drelu_grad_v2.cc37 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() function
58 return abstract::MakeAbstract(InferShape(input_args), InferType(primitive, input_args)); in ReLUGradV2Infer()

12345