| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | node_out_shapes.cc | 42 if (utils::isa<abstract::ShapePtr>(in_shape)) { in Run()
43 const auto &shape = in_shape->cast<abstract::ShapePtr>()->shape(); in Run()
58 if (!utils::isa<abstract::ShapePtr>(shape_ptr)) { in Run()
62 output_shapes.push_back(shape_ptr->cast<abstract::ShapePtr>()->shape()); in Run()
64 } else if (utils::isa<abstract::ShapePtr>(out_shape)) { in Run()
65 const auto &shape = out_shape->cast<abstract::ShapePtr>()->shape(); in Run()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | utils.h | 35 ShapePtr ShapeJoin(const ShapePtr &shape1, const ShapePtr &shape2);
58 ShapePtr GetBroadcastShape(const std::string &op, const AbstractTensorPtr &tensor_x, const Abstract…
67 AbstractBasePtr MakeAbstractTensor(const ShapePtr &shape, const TypePtr &type);
|
| D | utils.cc | 55 ShapePtr CalculateDynamicShape(const ShapePtr &shape1, const ShapePtr &shape2, const ShapeVector &d… in CalculateDynamicShape()
104 ShapePtr ShapeJoin(const ShapePtr &shape1, const ShapePtr &shape2) { in ShapeJoin()
344 AbstractBasePtr MakeAbstractTensor(const ShapePtr &shape, const TypePtr &type) { in MakeAbstractTensor()
385 auto shape = base_shape->cast<ShapePtr>(); in MakeAbstract()
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_batch_norm.cc | 52 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
56 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
59 auto shape3 = shape_vec[2]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
62 auto shape4 = shape_vec[3]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
65 auto shape5 = shape_vec[4]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_batch_norm_grad.cc | 56 auto shape0 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
59 auto shape1 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
62 auto shape2 = shape_vec[2]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
65 auto shape3 = shape_vec[3]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
68 auto shape4 = shape_vec[4]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_detection_post_process.cc | 50 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
55 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
59 auto shape3 = shape_vec[2]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
63 auto shape4 = shape_vec[3]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_softmax.cc | 49 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
79 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
109 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
|
| D | test_ops_lsh_projection.cc | 49 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
78 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
109 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
|
| D | test_ops_full_connection.cc | 48 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
79 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
109 auto shape = shape_ptr->cast<abstract::ShapePtr>(); in TEST_F()
|
| D | test_ops_unpack.cc | 51 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
54 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_unstack.cc | 51 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
54 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_custom_predict.cc | 52 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
55 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_merge.cc | 53 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
57 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_hashtable_lookup.cc | 54 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
56 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_layer_norm_beta_gamma_backprop_v2.cc | 55 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
58 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| D | test_ops_topk.cc | 64 auto shape1 = shape_vec[0]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
67 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf_utils.cc | 20 bool AnfUtils::IsShapeDynamic(const abstract::ShapePtr &shape) { in IsShapeDynamic()
36 …if (base_shape->isa<abstract::Shape>() && IsShapeDynamic(base_shape->cast<abstract::ShapePtr>())) { in IsNodeOutputDynamicShape()
43 if (b_shape->isa<abstract::Shape>() && IsShapeDynamic(b_shape->cast<abstract::ShapePtr>())) { in IsNodeOutputDynamicShape()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | reshape_eliminate.h | 34 using abstract::ShapePtr;
60 auto elements = tgt_shape->cast<ShapePtr>(); in operator()
61 auto shape = src_shape->cast<ShapePtr>(); in operator()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | addn.cc | 28 abstract::ShapePtr AddNInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr>… in AddNInferShape()
44 const auto &shape_vec = shape->cast<abstract::ShapePtr>()->shape(); in AddNInferShape()
45 const auto &shape_0_vec = shape_0->cast<abstract::ShapePtr>()->shape(); in AddNInferShape()
|
| D | leaky_relu.cc | 22 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
25 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape()
|
| D | relu.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
37 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape()
|
| D | relu6.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
38 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape()
|
| D | prelu.cc | 23 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
38 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | relu_grad_v2.cc | 30 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape()
33 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape()
|
| D | relu_grad.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
43 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape()
|