| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | split_with_size.cc | 26 std::vector<int64_t> split_size = GetArrayValue<int64_t>(input_abs).value().ToVector(); in CaculateSplitSize() local
50 auto split_size = CaculateSplitSize(input_args[kIndex1]); in InferShape() local
69 auto split_size = CaculateSplitSize(input_args[kIndex1]); in InferType() local
|
| D | split_tensor.cc | 26 auto split_size = GetScalarValue<int64_t>(split_size_value); in CaculateSplitSections() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | chunk.cc | 51 int64_t split_size = (dim_size + chunks_imm - 1) / chunks_imm; in ChunkAscendCustomize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | split_base.c | 49 int split_size = split_sizes[split_which]; in DoSplit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/ |
| D | split_tensor.cc | 58 int64_t split_size = input_shape[pos] / split_sections; in InferAbstract() local
|
| D | split_with_size.cc | 44 auto split_size = split_size_opt.value(); in InferAbstract() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/view/ |
| D | split_tensor_strides_calc.cc | 41 auto split_size = GetValue<int64_t>(inputs[kInputIndex1]); in SplitTensorCalc() local
|
| D | split_with_size_strides_calc.cc | 45 auto split_size = GetValue<std::vector<int64_t>>(inputs[kInputIndex1]); in SplitWithSizeCalc() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | split_base.cc | 77 int split_size = input_shape.at(param->split_dim_) / param->num_split_; in CheckAndInitSplitParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | split_cpu_kernel.cc | 87 int split_size = input_shape_[param.split_dim_] / SizeToInt(output_num_); in LaunchSplit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | inference_matmul_split_fusion.cc | 164 auto split_size = split_cnode->input(kIndex2)->cast<ValueNodePtr>(); in GetSplitSizeLen() local
188 auto split_size = split_cnode->input(kIndex2)->cast<ValueNodePtr>(); in CreateMatmulSplitPrim() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/expander/fallback/ |
| D | array_ops.cc | 247 …t SplitWithSizeFallbackFunc(FallbackIRBuilder *ib, const NodePtr &input, const NodePtr &split_size, in SplitWithSizeFallbackFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | split_fission.cc | 150 … int64_t split_size = std::accumulate(size_splits_new.begin(), size_splits_new.end(), int64_t(0)); in DoFission() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | resizebilinear_v2_info.cc | 437 Attr split_size = {SPLIT_SIZE, MakeValue(slice_size_)}; in CreateParallelResizeBilinearAttrs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_transform.cc | 155 auto split_size = ((*transform_op_list)[i - kIndex1].second)[kIndex1]; in OptimizeAllConcat() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 439 bool UpdateRatioWithPadStride(int64_t *ratio, size_t ratio_len, size_t split_size, int split_dim_si… in UpdateRatioWithPadStride()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _inner_ops.py | 1444 def __init__(self, ori_image_size, split_size, src_start_w, dst_start_w, align_corners): argument
|