| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/ |
| D | op_node.cc | 258 DShape ElemwiseOp::InferShape(const NodePtrList &inputs, const DAttrs &) { in InferShape() function in mindspore::opt::graphkernel::ElemwiseOp
311 DShape ReshapeOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::ReshapeOp
329 DShape BroadcastToOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::BroadcastToOp
347 DShape ReduceOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::ReduceOp
372 DShape Conv2dOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::Conv2dOp
435 DShape TransposeOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::TransposeOp
463 DShape MatMulOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::MatMulOp
494 DShape PadAkgOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::PadAkgOp
512 DShape UnPadAkgOp::InferShape(const NodePtrList &inputs, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::UnPadAkgOp
536 DShape StandardNormalOp::InferShape(const NodePtrList &, const DAttrs &attrs) { in InferShape() function in mindspore::opt::graphkernel::StandardNormalOp
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | infer.h | 25 typedef int (*InferShape)(const TensorC *const *inputs, size_t input_size, TensorC **outputs, size_… typedef
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | fft_imag.cc | 25 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | round.cc | 25 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape() function
|
| D | hshrink.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | assign_add.cc | 26 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | floor.cc | 29 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape() function
|
| D | div.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | sin.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | pow.cc | 25 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | less.cc | 28 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | maximum.cc | 26 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | ones_like.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | less_equal.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | logical_and.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | logical_or.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | squared_difference.cc | 28 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | rsqrt.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | zeros_like.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | hsigmoid.cc | 24 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | minimum.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | sub.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | soft_shrink.cc | 31 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | mul.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|
| D | relu.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
|