| /third_party/mindspore/mindspore/core/ops/ |
| D | split_v.cc | 42 auto size_splits = GetValue<std::vector<int64_t>>(primitive->GetAttr(kSizeSplits)); in InferShape() local
43 …Utils::Check("num_split", num_split, kEqual, "rank of size_splits", SizeToLong(size_splits.size()), in InferShape()
45 auto default_idx = std::find(size_splits.begin(), size_splits.end(), -1); in InferShape()
46 if (default_idx == size_splits.end()) { in InferShape()
49 …CheckAndConvertUtils::CheckInRange("elements of size_splits", size_splits[LongToSize(i)], kInclude… in InferShape()
51 sum_of_size_splits += size_splits[LongToSize(i)]; in InferShape()
56 (void)size_splits.erase(default_idx); in InferShape()
57 auto excessive_default_idx = std::find(size_splits.begin(), size_splits.end(), -1); in InferShape()
58 if (excessive_default_idx != size_splits.end()) { in InferShape()
63 …CheckAndConvertUtils::CheckInRange("elements of size_splits", size_splits[LongToSize(i)], kInclude… in InferShape()
[all …]
|
| D | split.cc | 29 void Split::set_size_splits(const std::vector<int64_t> &size_splits) { in set_size_splits() argument
30 (void)this->AddAttr(kSizeSplits, MakeValue(size_splits)); in set_size_splits()
|
| D | split.h | 41 void set_size_splits(const std::vector<int64_t> &size_splits);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_slice_parser.cc | 32 std::vector<int64_t> size_splits; in Parse() local
35 size_splits.push_back(slice_param.slice_point(i)); in Parse()
37 size_splits.push_back(slice_param.slice_point(i) - slice_param.slice_point(i - 1)); in Parse()
40 size_splits.push_back(-1); in Parse()
41 prim->set_size_splits(size_splits); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_split_parser.cc | 30 std::vector<int64_t> size_splits; in Parse() local
37 size_splits.resize(onnx_node_attr.ints_size()); in Parse()
38 std::copy(onnx_node_attr.ints().begin(), onnx_node_attr.ints().end(), size_splits.begin()); in Parse()
39 prim->set_size_splits(size_splits); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_split_v_parser.cc | 43 std::vector<int64_t> size_splits; in Parse() local
44 …tTfliteData(tflite_op->inputs[1], tflite_subgraph->tensors, tflite_model->buffers, &size_splits)) { in Parse()
48 prim->set_size_splits(size_splits); in Parse()
|
| D | tflite_split_parser.cc | 74 std::vector<int64_t> size_splits; in Parse() local
77 size_splits.push_back(tensor_shape[axis] / num_splits); in Parse()
80 prim->set_size_splits(size_splits); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_split_parser.cc | 84 std::vector<int64_t> size_splits; in Parse() local
85 …std::transform(size_splits_int32.begin(), size_splits_int32.end(), std::back_inserter(size_splits), in Parse()
87 prim->set_size_splits(size_splits); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | multi_conv_info.cc | 49 split_info_.size_splits.clear(); in GenSplitInfo()
50 split_info_.size_splits = this->strategy_.strategys[0][i]; // cal base on compute_cap in GenSplitInfo()
55 split_info_.extend_bottom = std::vector<int64_t>(split_info_.size_splits.size(), 0); in GenSplitInfo()
56 split_info_.extend_top = std::vector<int64_t>(split_info_.size_splits.size(), 0); in GenSplitInfo()
58 ori_split_ratios_ = split_info_.size_splits; in GenSplitInfo()
78 auto final_ratios = split_info_.size_splits; in CheckSplitValid()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 92 split_len += split_info->size_splits[i]; in CalSplitOutputShape()
102 if (INT_MUL_OVERFLOW_THRESHOLD(split_info->size_splits[i], splited_axis_value, INT64_MAX)) { in CalSplitOutputShape()
106 int64_t tmp = UP_DIV(split_info->size_splits[i] * splited_axis_value, split_len); in CalSplitOutputShape()
248 size_t split_num = split_info->size_splits.size(); in UpdateSplitInfo()
273 split_info->size_splits.clear(); in UpdateSplitInfo()
279 split_info->size_splits.push_back(split_axis_inputs_shape[node_size][0]); in UpdateSplitInfo()
287 split_info->size_splits.push_back(value); in UpdateSplitInfo()
370 split_prim->set_ratio(split_info->size_splits); in CreateOutputsOfSplitWithOverlap()
|
| D | fisson_util.h | 39 std::vector<int64_t> size_splits{};
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_inputs_for_reduce_scatter.cc | 36 std::vector<int> size_splits; in InsertSplitForInput() local
41 size_splits.push_back(output_node_shape[0]); in InsertSplitForInput()
46 AnfAlgo::SetNodeAttr("size_splits", MakeValue(size_splits), split); in InsertSplitForInput()
|
| /third_party/mindspore/mindspore/lite/test/ut/tools/converter/parser/tflite/ |
| D | tflite_split_parser_test.cc | 41 ASSERT_EQ(val->size_splits, sizeSplits); in TEST_F()
|
| D | tflite_split_v_parser_test.cc | 41 ASSERT_EQ(val->size_splits, sizeSplits); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | split_populate.cc | 65 auto split_sizes_vector_ = value->size_splits(); in PopulateSplitParameter()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | split_fission.cc | 41 void SetAttrForSplitVNode(const AnfNodePtr &splitv, const std::vector<int64_t> &size_splits, int64_… in SetAttrForSplitVNode() argument
43 AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(size_splits), splitv); in SetAttrForSplitVNode()
|
| D | dynamic_gru_v2_grad_fission.cc | 328 std::vector<int64_t> size_splits = {SizeToLong(hidden_size << 1), SizeToLong(hidden_size)}; in CreateDgateHSplitVDNode() local
329 AnfAlgo::SetNodeAttr("size_splits", MakeValue(size_splits), split_vd); in CreateDgateHSplitVDNode()
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | split_npu.cc | 35 auto sizes_split = split_prim->size_splits(); in Init()
|
| /third_party/mindspore/mindspore/lite/mindir/include/ |
| D | mindir.h | 358 …R_Split_CreatePrimitive(int64_t output_num, const std::vector<int64_t> &size_splits, int64_t axis);
362 void MindIR_Split_SetSizeSplits(PrimitivePtr *primitive, const std::vector<int64_t> &size_splits);
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir_nnrt_lite_graph_to_model.cc | 1111 …if (value->size_splits() == nullptr || value->size_splits()->size() <= static_cast<uint32_t>(split… in ConvertSplit()
1114 … sizeSplits = std::vector<int64_t>(value->size_splits()->begin(), value->size_splits()->end()); in ConvertSplit()
|
| D | mindir.cc | 3651 …_Split_CreatePrimitive(int64_t output_num, const std::vector<int64_t> &size_splits, int64_t axis) { in MindIR_Split_CreatePrimitive() argument
3654 …schema::CreateSplit(fbb, output_num, fbb.CreateVector(size_splits.data(), size_splits.size()), axi… in MindIR_Split_CreatePrimitive()
3682 …fbb, output_num, fbb.CreateVector(value->size_splits()->data(), value->size_splits()->size()), val… in MindIR_Split_SetOutputNum()
3698 auto src = value->size_splits(); in MindIR_Split_GetSizeSplits()
3710 void MindIR_Split_SetSizeSplits(PrimitivePtr *primitive, const std::vector<int64_t> &size_splits) { in MindIR_Split_SetSizeSplits() argument
3717 … fbb.CreateVector(size_splits.data(), size_splits.size()), value->axis()); in MindIR_Split_SetSizeSplits()
3748 …fbb, value->output_num(), fbb.CreateVector(value->size_splits()->data(), value->size_splits()->siz… in MindIR_Split_SetAxis()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 1087 size_splits = [x_shape[self.axis] // self.output_num] * self.output_num
1088 self.add_prim_attr('size_splits', size_splits)
6504 def __init__(self, size_splits, split_dim, num_split): argument
6506 validator.check_value_type("size_splits", size_splits, [tuple, list], self.name)
6507 for elements_of_size_splits in size_splits:
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops.fbs | 947 size_splits: [long];
|
| D | ops_generated.h | 9838 const flatbuffers::Vector<int64_t> *size_splits() const {
9848 verifier.VerifyVector(size_splits()) &&
9861 void add_size_splits(flatbuffers::Offset<flatbuffers::Vector<int64_t>> size_splits) {
9862 fbb_.AddOffset(Split::VT_SIZE_SPLITS, size_splits);
9881 flatbuffers::Offset<flatbuffers::Vector<int64_t>> size_splits = 0,
9886 builder_.add_size_splits(size_splits);
9893 const std::vector<int64_t> *size_splits = nullptr,
9895 auto size_splits__ = size_splits ? _fbb.CreateVector<int64_t>(*size_splits) : 0;
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_def.cc | 947 OP_ATTR(size_splits, [long])
|