Home
last modified time | relevance | path

Searched refs:TupleShapePtr (Results 1 – 25 of 30) sorted by relevance

12

/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dnode_out_shapes.cc54 if (utils::isa<abstract::TupleShapePtr>(out_shape)) { in Run()
55 auto shape = out_shape->cast<abstract::TupleShapePtr>(); in Run()
/third_party/mindspore/mindspore/core/ir/
Danf_utils.cc39 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in IsNodeOutputDynamicShape()
/third_party/mindspore/mindspore/core/ops/
Dmerge.cc33 auto inputs_shape = input_args[0]->BuildShape()->cast<abstract::TupleShapePtr>()->shape(); in MergeInfer()
Dlayer_norm_x_backprop_v2.cc28 abstract::TupleShapePtr LayerNormXBackpropV2InferShape(const PrimitivePtr &primitive, in LayerNormXBackpropV2InferShape()
DLayerNormBetaGammaBackprop.cc28 abstract::TupleShapePtr LayerNormBetaGammaBackpropInferShape(const PrimitivePtr &primitive, in LayerNormBetaGammaBackpropInferShape()
Dlayer_norm_beta_gamma_backprop_v2.cc28 abstract::TupleShapePtr LayerNormBetaGammaBackpropV2InferShape(const PrimitivePtr &primitive) { in LayerNormBetaGammaBackpropV2InferShape()
Ddynamic_broadcast_gradient_args.cc55 abstract::TupleShapePtr Infer(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in Infer()
Dctc_loss_v2.cc34 abstract::TupleShapePtr CTCLossV2InferShape(const PrimitivePtr &primitive, in CTCLossV2InferShape()
Dpack.cc62 auto x_shapes = input_args[0]->BuildShape()->cast<abstract::TupleShapePtr>()->shape(); in PackInfer()
Dreluv2.cc56 abstract::TupleShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr… in InferShape()
Dsparse_apply_r_m_s_prop.cc30 abstract::TupleShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr… in InferShape()
Dctcloss.cc69 abstract::TupleShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr… in InferShape()
Dsplit_v.cc27 abstract::TupleShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr… in InferShape()
Dapply_adagrad_d_a.cc30 abstract::TupleShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr… in InferShape()
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_unstack.cc47 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_unpack.cc47 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_custom_predict.cc48 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_hashtable_lookup.cc50 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_merge.cc49 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_layer_norm_beta_gamma_backprop_v2.cc51 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_topk.cc60 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_layer_norm_x_backprop_v2.cc54 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_detection_post_process.cc46 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_batch_norm.cc48 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()
Dtest_ops_batch_norm_grad.cc52 auto shape = shape_ptr->cast<abstract::TupleShapePtr>(); in TEST_F()

12