| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | split_fission.cc | 42 int64_t num_split) { in SetAttrForSplitVNode() argument
45 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(num_split), splitv); in SetAttrForSplitVNode()
48 size_t GetSmallSplitSize(const AnfNodePtr &split_node, int64_t split_dim, int64_t num_split) { in GetSmallSplitSize() argument
56 if (num_split == 0) { in GetSmallSplitSize()
59 return input_shape[LongToSize(split_dim)] / LongToSize(num_split); in GetSmallSplitSize()
81 …peAndTypeId(const CNodePtr &origin_cnode, int64_t split_dim, int64_t split_size, int64_t num_split, in CreateOutputShapeAndTypeId() argument
92 for (int64_t i = 0; i < num_split; ++i) { in CreateOutputShapeAndTypeId()
101 int64_t num_split) { in SetAttrAndAbstractForBaseSplitv() argument
102 SetAttrForSplitVNode(base_splitv, size_splits_base, split_dim, num_split); in SetAttrAndAbstractForBaseSplitv()
105 std::vector<TypeId> base_type_ids(num_split, type_id); in SetAttrAndAbstractForBaseSplitv()
[all …]
|
| D | splitv_fission.cc | 38 auto num_split = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrNumSplit); in Process() local
39 if (num_split <= outputs_divisor_) { in Process()
42 …return DoFission(func_graph, cnode, num_split, outputs_divisor_, AnfAlgo::GetNodeAttr<int64_t>(cno… in Process()
|
| D | split_fission.h | 35 …AnfNodePtr DoFission(const FuncGraphPtr &func_graph, const CNodePtr &cnode, int64_t num_split, int…
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | split_v.cc | 40 auto num_split = GetValue<int64_t>(primitive->GetAttr("num_split")); in InferShape() local
41 (void)CheckAndConvertUtils::CheckInteger("num_split", num_split, kGreaterEqual, 1, prim_name); in InferShape()
43 …CheckAndConvertUtils::Check("num_split", num_split, kEqual, "rank of size_splits", SizeToLong(size… in InferShape()
48 for (int64_t i = 0; i < num_split; i++) { in InferShape()
62 for (int64_t i = 0; i < num_split - 1; i++) { in InferShape()
72 for (int64_t i = 0; i < num_split; i++) { in InferShape()
85 auto num_split = GetValue<int64_t>(prim->GetAttr("num_split")); in InferType() local
92 for (int64_t i = 0; i < num_split; i++) { in InferType()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | split_base.c | 26 const int num_split = split_param->num_split_; in DoSplit() local
38 split_which = offset % num_split; in DoSplit()
39 split_times = offset / num_split; in DoSplit()
47 split_which = i % num_split; in DoSplit()
48 split_times = i / num_split; in DoSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | split_int8.c | 27 const int num_split = param->num_split_; in Int8DoSplit() local
34 int split_which = offset % num_split; in Int8DoSplit()
35 int split_times = offset / num_split; in Int8DoSplit()
47 split_which = i % num_split; in Int8DoSplit()
48 split_times = i / num_split; in Int8DoSplit()
|
| /third_party/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_pack.c | 650 int num_split = src_type.width * src_type.length / 128; in lp_build_pack2() local
662 assert(num_split <= LP_MAX_VECTOR_WIDTH / 128); in lp_build_pack2()
664 for (i = 0; i < num_split / 2; i++) { in lp_build_pack2()
675 for (i = 0; i < num_split / 2; i++) { in lp_build_pack2()
680 tmpres[i+num_split/2] = lp_build_intrinsic_binary(builder, intrinsic, in lp_build_pack2()
684 tmpres[i+num_split/2] = LLVMBuildBitCast(builder, tmpres[i+num_split/2], in lp_build_pack2()
688 res = lp_build_concat(gallivm, tmpres, ndst_type, num_split); in lp_build_pack2()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | split.cc | 111 size_t num_split = UP_DIV(shape_dim, param->split_sizes_[0]); in AlignSplitSizes() local
112 …split_sizes_ = reinterpret_cast<int *>(allocator->Malloc(num_split * sizeof(int), lite::opencl::Me… in AlignSplitSizes()
117 for (int i = 0; i < num_split - 1; ++i) { in AlignSplitSizes()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | split_combination_ops_declare.cc | 24 {"output_num", ATTR_DESC(num_split, AnyTraits<int64_t>())}};
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | depthwise_conv2d_info.cc | 38 void SplitConstantData(char *in_data, char **out_data, int64_t num_split, int64_t split_dim_size, i… in SplitConstantData() argument
43 for (int64_t slice_idx = 0; slice_idx < num_split; slice_idx++) { in SplitConstantData()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 6504 def __init__(self, size_splits, split_dim, num_split): argument
6513 validator.check_value_type("num_split", num_split, [int], self.name)
6514 validator.check_positive_int(num_split, "num_split", self.name)
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 2613 'block': P.SplitV(size_splits=[1, 2], split_dim=1, num_split=2),
|
| /third_party/mindspore/config/ |
| D | op_info.config | 473 …m", "param_type": "required", "type": "int", "value": "all"}, {"name": "num_split", "param_type": …
|