| /third_party/mindspore/mindspore/core/ops/ |
| D | zeros.cc | 29 …pePtr ZerosInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in ZerosInferShape()
39 TypePtr ZerosInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in ZerosInferType()
53 const std::vector<AbstractBasePtr> &input_args) { in ZerosInfer()
58 ValuePtr ZerosInferValue(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in ZerosInferValue()
|
| D | round.cc | 25 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape()
30 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
38 const std::vector<AbstractBasePtr> &input_args) { in RoundInfer()
|
| D | ones.cc | 29 …apePtr OnesInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in OnesInferShape()
38 TypePtr OnesInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in OnesInferType()
51 const std::vector<AbstractBasePtr> &input_args) { in OnesInfer()
60 ValuePtr OnesInferValue(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in OnesInferValue()
|
| D | hshrink.cc | 29 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
35 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
45 const std::vector<AbstractBasePtr> &input_args) { in HShrinkInfer()
|
| D | assign_add.cc | 26 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
32 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
42 const std::vector<AbstractBasePtr> &input_args) { in AssignAddInfer()
|
| D | sin.cc | 30 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
39 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
49 const std::vector<AbstractBasePtr> &input_args) { in SinInfer()
|
| D | div.cc | 29 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
35 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
47 const std::vector<AbstractBasePtr> &input_args) { in DivInfer()
|
| D | less.cc | 28 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
34 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
47 const std::vector<AbstractBasePtr> &input_args) { in LessInfer()
|
| D | maximum.cc | 26 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
32 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
44 const std::vector<AbstractBasePtr> &input_args) { in MaximumInfer()
|
| D | floor.cc | 29 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape()
34 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
42 const std::vector<AbstractBasePtr> &input_args) { in FloorInfer()
|
| D | pow.cc | 25 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
31 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
43 const std::vector<AbstractBasePtr> &input_args) { in PowInfer()
|
| D | cos.cc | 25 …hapePtr CosInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in CosInferShape()
31 TypePtr CosInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in CosInferType()
38 const std::vector<AbstractBasePtr> &input_args) { in CosInfer()
|
| D | less_equal.cc | 29 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
35 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
47 const std::vector<AbstractBasePtr> &input_args) { in LessEqualInfer()
|
| D | logical_and.cc | 29 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
35 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
48 const std::vector<AbstractBasePtr> &input_args) { in LogicalAndInfer()
|
| D | squared_difference.cc | 28 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
33 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
43 const std::vector<AbstractBasePtr> &input_args) { in SquaredDifferenceInfer()
|
| D | logical_or.cc | 30 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
36 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
46 const std::vector<AbstractBasePtr> &input_args) { in LogicalOrInfer()
|
| D | logical_not.cc | 26 … LogicalNotInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in LogicalNotInferShape()
32 …ypePtr LogicalNotInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in LogicalNotInferType()
42 const std::vector<AbstractBasePtr> &input_args) { in LogicalNotInfer()
|
| D | ones_like.cc | 30 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
48 const std::vector<AbstractBasePtr> &input_args) { in OnesLikeInfer()
|
| D | diag.cc | 27 …apePtr DiagInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in DiagInferShape()
37 TypePtr PartInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args… in PartInferType()
44 const std::vector<AbstractBasePtr> &input_args) { in DiagInfer()
|
| D | rsqrt.cc | 30 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
40 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
49 const std::vector<AbstractBasePtr> &input_args) { in RsqrtInfer()
|
| D | zeros_like.cc | 30 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
46 const std::vector<AbstractBasePtr> &input_args) { in ZerosLikeInfer()
|
| D | layer_norm_x_backprop_v2.cc | 29 … const std::vector<AbstractBasePtr> &input_args) { in LayerNormXBackpropV2InferShape()
36 …erNormXBackpropV2InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_arg… in LayerNormXBackpropV2InferType()
48 const std::vector<AbstractBasePtr> &input_args) { in LayerNormXBackpropV2Infer()
|
| D | custom_normalize.cc | 24 abstract::ShapePtr CustomNormalizeInferShape(const std::vector<AbstractBasePtr> &input_args) { in CustomNormalizeInferShape()
40 TypePtr CustomNormalizeInferType(const std::vector<AbstractBasePtr> &input_args) { in CustomNormalizeInferType()
51 const std::vector<AbstractBasePtr> &input_args) { in CustomNormalizeInfer()
|
| D | hsigmoid.cc | 24 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
33 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
44 const std::vector<AbstractBasePtr> &input_args) { in HSigmoidInfer()
|
| D | minimum.cc | 30 …::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_arg… in InferShape()
36 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
52 const std::vector<AbstractBasePtr> &input_args) { in MinimumInfer()
|