Home
last modified time | relevance | path

Searched defs:shape_list (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Ddropout_gen_mask.cc66 ShapeVector CalOutputShape(const AbstractBasePtrList shape_list) { in CalOutputShape()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_util.cc122 const vector<vector<size_t>> &shape_list, uint64_t *total_count) { in GetHcomCount()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/
Dpybind_conversion.cc297 py::list shape_list; in shapesToListOfShape() local
/third_party/mindspore/mindspore/core/abstract/
Dutils.cc405 auto shape_list = base_shape->cast<ListShapePtr>(); in MakeAbstract() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc303 auto shape_list = ExtractShape(temp_node); in CreateOpInfo() local
502 auto shape_list = std::make_shared<ValueList>(element); in GetShapeType() local
/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils_py.cc476 auto shape_list = shape_obj.cast<py::list>(); in MakePyInferRes2Abstract() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc1256 const std::vector<Shapes> &shape_list) { in OperatorInstanceByName()
1293 const std::vector<Shapes> &shape_list) { in OperatorInstance()
1308 std::vector<Shapes> shape_list) { in NewOperatorInstance()
1762 std::vector<Shapes> shape_list = ExtractShape(node); in SetVirtualDatasetStrategy() local
1896 std::vector<Shapes> shape_list = ExtractShape(cnode); in ExtractInformation() local
Dstep_auto_parallel.cc296 std::vector<Shapes> shape_list = ExtractShape(cnode); in CreateTheOperatorInfo() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc92 std::vector<Shapes> shape_list = ExtractShape(cnode); in SetStridedSliceStrategy() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprim.cc366 py::list shape_list(len); in AbstractListToPython() local
/third_party/mindspore/mindspore/ccsrc/common/
Dtrans.cc221 bool HasShapeDynamic(const std::vector<int64_t> &shape_list) { in HasShapeDynamic()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc2005 auto shape_list = sequeue_shape_ptr->shape(); in GetShapeFromSequeueShape() local