| /third_party/mindspore/mindspore/compression/quant/ |
| D | quant_utils.py | 113 shape_list = [-1] + [1] * len(data.shape[1:])
114 scale = scale.reshape(shape_list)
115 zero_point = zero_point.reshape(shape_list)
118 shape_list = [1, -1] + [1] * len(data.shape[2:])
119 scale = scale.reshape(shape_list)
120 zero_point = zero_point.reshape(shape_list)
170 shape_list = [-1] + [1] * len(weight.shape[1:])
171 _gamma = gamma.reshape(shape_list)
172 _sigma = sigma.reshape(shape_list)
175 shape_list = [1, -1] + [1] * len(weight.shape[2:])
[all …]
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | dropout_gen_mask.cc | 66 ShapeVector CalOutputShape(const AbstractBasePtrList shape_list) { in CalOutputShape() argument
68 size_t x_rank = shape_list.size(); in CalOutputShape()
70 auto value_track = shape_list[i]->GetValueTrack(); in CalOutputShape()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 92 const std::vector<Shapes> &shape_list);
96 std::vector<Shapes> shape_list);
|
| D | step_parallel.cc | 1256 const std::vector<Shapes> &shape_list) { in OperatorInstanceByName() argument
1257 if (shape_list.size() != 2) { in OperatorInstanceByName()
1280 …(OperatorInfoPtr)DynCreator::Instance().Create(distribute_opname, shape_list[0], shape_list[1], at… in OperatorInstanceByName()
1293 const std::vector<Shapes> &shape_list) { in OperatorInstance() argument
1295 OperatorInfoPtr operator_ = OperatorInstanceByName(prim->name(), attrs, shape_list); in OperatorInstance()
1301 operator_ = OperatorInstanceByName(BATCH_PARALLEL, attrs, shape_list); in OperatorInstance()
1308 std::vector<Shapes> shape_list) { in NewOperatorInstance() argument
1309 OperatorInfoPtr operator_ = OperatorInstance(prim, attrs, shape_list); in NewOperatorInstance()
1310 for (size_t i = 0; i < shape_list[0].size(); ++i) { in NewOperatorInstance()
1311 MS_LOG(INFO) << "No: " << i << " input's shape: " << ShapeToString(shape_list[0][i]); in NewOperatorInstance()
[all …]
|
| D | step_auto_parallel.cc | 296 std::vector<Shapes> shape_list = ExtractShape(cnode); in CreateTheOperatorInfo() local
297 if (shape_list.empty()) { in CreateTheOperatorInfo()
301 OperatorInfoPtr operator_info = NewOperatorInstance(prim, attrs, shape_list); in CreateTheOperatorInfo()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/ |
| D | pybind_conversion.cc | 297 py::list shape_list; in shapesToListOfShape() local
299 shape_list.append(shape.AsVector()); in shapesToListOfShape()
301 return shape_list; in shapesToListOfShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_util.h | 67 const vector<vector<size_t>> &shape_list, uint64_t *total_count);
|
| D | hcom_util.cc | 122 const vector<vector<size_t>> &shape_list, uint64_t *total_count) { in GetHcomCount() argument
142 if (!GetHcclOpSize(data_type_list[i], shape_list[i], &input_size)) { in GetHcomCount()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | utils.cc | 405 auto shape_list = base_shape->cast<ListShapePtr>(); in MakeAbstract() local
408 for (size_t it = 0; it < shape_list->size(); ++it) { in MakeAbstract()
409 auto tensor_it = MakeAbstract((*shape_list)[it], (*type_list)[it]); in MakeAbstract()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 92 std::vector<Shapes> shape_list = ExtractShape(cnode); in SetStridedSliceStrategy() local
93 if (shape_list.empty()) { in SetStridedSliceStrategy()
97 for (size_t i = 0; i < shape_list[0].size(); i++) { in SetStridedSliceStrategy()
98 if (shape_list[0][i].empty()) { in SetStridedSliceStrategy()
102 for (size_t j = 0; j < shape_list[0][i].size(); j++) { in SetStridedSliceStrategy()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 303 auto shape_list = ExtractShape(temp_node); in CreateOpInfo() local
304 if (shape_list.empty()) { in CreateOpInfo()
313 auto op_info = OperatorInstance(prim, attrs, shape_list); in CreateOpInfo()
502 auto shape_list = std::make_shared<ValueList>(element); in GetShapeType() local
507 return std::make_pair(shape_list, dtype); in GetShapeType()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | convert_utils_py.cc | 476 auto shape_list = shape_obj.cast<py::list>(); in MakePyInferRes2Abstract() local
479 for (size_t it = 0; it < shape_list.size(); ++it) { in MakePyInferRes2Abstract()
480 auto tensor_it = MakePyInferRes2Abstract(shape_list[it], typeid_list[it]); in MakePyInferRes2Abstract()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.cc | 366 py::list shape_list(len); in AbstractListToPython() local
375 shape_list[i] = out[ATTR_SHAPE]; in AbstractListToPython()
387 dic[ATTR_SHAPE] = shape_list; in AbstractListToPython()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2005 auto shape_list = sequeue_shape_ptr->shape(); in GetShapeFromSequeueShape() local
2006 if (index >= shape_list.size()) { in GetShapeFromSequeueShape()
2007 MS_LOG(EXCEPTION) << "Output Index:" << index << " >= " << shape_list.size(); in GetShapeFromSequeueShape()
2010 auto shape = shape_list[index]; in GetShapeFromSequeueShape()
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 221 bool HasShapeDynamic(const std::vector<int64_t> &shape_list) { in HasShapeDynamic() argument
222 …return std::any_of(shape_list.begin(), shape_list.end(), [](int64_t shape) { return shape == Shape… in HasShapeDynamic()
|