Home
last modified time | relevance | path

Searched defs:dynamic_shape (Results 1 – 25 of 36) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dsplit_with_size.cc38 …auto dynamic_shape = std::make_shared<abstract::Shape>(std::vector<int64_t>{abstract::Shape::kShap… in InferAbstract() local
46 …auto dynamic_shape = std::make_shared<abstract::Shape>(std::vector<int64_t>{abstract::Shape::kShap… in InferAbstract() local
51 auto dynamic_shape = in InferAbstract() local
Dsplit_tensor.cc42 …auto dynamic_shape = std::make_shared<abstract::Shape>(std::vector<int64_t>{abstract::Shape::kShap… in InferAbstract() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_ps_roi_pooling_grad_v2_op.py30 def __init__(self, input_size, spatial_scale, group_size, output_dim, dynamic_shape=False): argument
45 def ps_roi_pooling_grad_case(data_type, dynamic_shape=False): argument
Dtest_dynamic_shape_op.py36 def dynamic_shape(np_type): function
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_ps_roi_pooling_v2_op.py38 def __init__(self, spatial_scale, group_size, output_dim, dynamic_shape=False): argument
53 def _ps_roi_pooling_case(data_type, mode, x_size_adjust=None, dynamic_shape=False): argument
Dtest_ps_roi_pooling_grad_op.py35 def __init__(self, input_size, spatial_scale, group_size, output_dim, dynamic_shape=False): argument
51 def _ps_roi_pooling_grad_case(data_type, mode, y_size_adjust=None, dynamic_shape=False): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/
Dtest_matrix_determinant.py33 def dynamic_shape(): function
Dtest_atan2.py33 def dynamic_shape(): function
Dtest_div.py33 def dynamic_shape(): function
Dtest_conv2dtranspose.py37 def dynamic_shape(): function
Dtest_addcdiv.py33 def dynamic_shape(): function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Dconvert_custom_op.cc28 namespace opt::dynamic_shape { namespace
Dlink_custom_op.cc31 namespace opt::dynamic_shape { namespace
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dmv.cc45 bool dynamic_shape = IsDynamic(x_shp) || IsDynamic(y_shp); in InferShape() local
Ddot.cc45 bool dynamic_shape = IsDynamic(x_shp) || IsDynamic(y_shp); in InferShape() local
Daddmm.cc45 bool dynamic_shape = IsDynamic(x_shp) || IsDynamic(y_shp); in InferShape() local
Dbmm_ext.cc60 bool dynamic_shape = IsDynamic(x_shp) || IsDynamic(y_shp); in InferShape() local
Dquant_batch_matmul.cc73 bool dynamic_shape = IsDynamic(x1_shp) || IsDynamic(x2_shp); in InferShape() local
Dmatmul_ext.cc153 bool dynamic_shape = IsDynamic(x_shp) || IsDynamic(y_shp); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/
Dconfig.h79 void set_dynamic_shape(bool dynamic_shape) { dynamic_shape_ = dynamic_shape; } in set_dynamic_shape()
80 bool dynamic_shape() const { return dynamic_shape_; } in dynamic_shape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Ddropout_for_ge.cc67 CNodePtr dynamic_shape = func_graph->NewCNode(dynamic_shape_inputs); in CreateDynamicShapeCNode() local
86 …CNodePtr dynamic_shape = CreateDynamicShapeCNode(func_graph, dropout->input(kIndex1), input_shape); in CreateDropoutGenMaskCNode() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/symbol_engine/ops/
Dtest_strided_slice.cc55 TEST_P(TestStridedSlice, dynamic_shape) { in TEST_P() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/
Dgrad.h78 inline DynamicShapePtr dynamic_shape() const { in dynamic_shape() function
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dfractional_max_pool_grad_with_fixed_ksize.cc60 std::vector<bool> dynamic_shape; in InputDynamic() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_data_queue.cc268 std::shared_ptr<DataQueue> CreateGpuDataQueue(const std::string &channel_name, bool dynamic_shape, … in CreateGpuDataQueue()

12