| /third_party/mindspore/mindspore/lite/src/runtime/ |
| D | infer_manager.h | 47 …void InsertInferShapeFunc(int prim_type, InferShape func) { infer_shape_funcs_[prim_type] = func; } in InsertInferShapeFunc()
49 InferShape GetInferShapeFunc(int prim_type) { in GetInferShapeFunc()
60 std::map<int, InferShape> infer_shape_funcs_;
65 RegistryInferShape(int prim_type, InferShape func) { in RegistryInferShape()
71 #define REG_INFER_SHAPE(op, prim_type, func) static RegistryInferShape g_##op##InferShape(prim_type…
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/ |
| D | op_node.h | 69 …virtual DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) { return inputs[0]->shap… in InferShape() function
86 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
105 …DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override { return inputs[2]->sha… in InferShape() function
165 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
178 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
188 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
204 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
214 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
224 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
234 DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override;
[all …]
|
| D | op_node.cc | 77 nodebase.shape = InferShape(inputs, attrs); in Infer()
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
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/expanders/ |
| D | reshape.cc | 32 static ShapeVector InferShape(const ShapeVector &shape, const std::vector<int64_t> &axis) { in InferShape() function in mindspore::opt::expanders::ExpandDims
52 … auto shape = MakeValue(ExpandDims::InferShape(input_x->shape, GetAxisList(this->attrs_["axis"]))); in Expand()
60 return ExpandDims::InferShape(shape, axis); in ExpandDimsInferShape()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | space_to_batch_nd_tests.cc | 39 std::vector<int> InferShape(const std::vector<int> &input_shape, const std::vector<int> &block_size… in InferShape() function
53 auto output_shape = InferShape(input_shape, block_sizes, paddings); in TEST_F()
92 auto output_shape = InferShape(input_shape, block_sizes, paddings); in TEST_F()
|
| /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
28 InferShape GetInferFunc(int prim_type);
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | round.cc | 25 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape() function
39 …return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), InferShape(inp… in RoundInfer()
|
| D | fft_imag.cc | 25 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
39 return std::make_shared<abstract::AbstractTensor>(kFloat32, InferShape(primitive, input_args)); in FftImagInfer()
|
| D | sin.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
51 InferShape(primitive, input_args)->shape()); in SinInfer()
|
| D | maximum.cc | 26 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
46 InferShape(primitive, input_args)->shape()); in MaximumInfer()
|
| D | div.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
49 InferShape(primitive, input_args)->shape()); in DivInfer()
|
| D | pow.cc | 25 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
45 InferShape(primitive, input_args)->shape()); in PowInfer()
|
| D | ones_like.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
50 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in OnesLikeInfer()
|
| D | less.cc | 28 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
49 InferShape(primitive, input_args)->shape()); in LessInfer()
|
| D | assign_add.cc | 26 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
44 InferShape(primitive, input_args)->shape()); in AssignAddInfer()
|
| D | less_equal.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
49 InferShape(primitive, input_args)->shape()); in LessEqualInfer()
|
| D | logical_and.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
50 InferShape(primitive, input_args)->shape()); in LogicalAndInfer()
|
| D | squared_difference.cc | 28 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
48 InferShape(primitive, input_args)); in SquaredDifferenceInfer()
|
| D | hshrink.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
47 InferShape(primitive, input_args)->shape()); in HShrinkInfer()
|
| D | floor.cc | 29 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape() function
46 …return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), InferShape(inp… in FloorInfer()
|
| D | logical_or.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
51 InferShape(primitive, input_args)); in LogicalOrInfer()
|
| D | zeros_like.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
50 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in ZerosLikeInfer()
|
| D | hsigmoid.cc | 24 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
46 InferShape(primitive, input_args)->shape()); in HSigmoidInfer()
|
| D | rsqrt.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() function
51 InferShape(primitive, input_args)->shape()); in RsqrtInfer()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.cc | 81 void DynamicKernel::InferShape() { in InferShape() function in mindspore::device::DynamicKernel
92 AnfAlgo::InferShape(cnode, &depend_tensor_map_); in InferShape()
134 AnfAlgo::InferShape(nop_node->cast<CNodePtr>()); in InferShapeForNopNode()
|