| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | matrix_split_impl.cu | 20 __global__ void MatrixSplitKernel(const size_t size, const size_t split_dim, const size_t dim, T *i… in MatrixSplitKernel()
32 __global__ void MatrixSplitKernel(const size_t size, const size_t split_dim, const size_t dim, cons… in MatrixSplitKernel()
55 void MatrixSplit(const size_t size, const size_t split_dim, const size_t dim, T *input_addr, T *out… in MatrixSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | split_fission.cc | 41 …ForSplitVNode(const AnfNodePtr &splitv, const std::vector<int64_t> &size_splits, int64_t split_dim, in SetAttrForSplitVNode()
48 size_t GetSmallSplitSize(const AnfNodePtr &split_node, int64_t split_dim, int64_t num_split) { in GetSmallSplitSize()
81 void CreateOutputShapeAndTypeId(const CNodePtr &origin_cnode, int64_t split_dim, int64_t split_size… in CreateOutputShapeAndTypeId()
100 … const std::vector<int64_t> &size_splits_base, int64_t split_dim, in SetAttrAndAbstractForBaseSplitv()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | split_with_overlap.cc | 23 int64_t split_dim, int64_t stride, int64_t pad_top, bool trans_format) { in Init()
48 void SplitWithOverlap::set_split_dim(int64_t split_dim) { (void)this->AddAttr(kSplitDim, MakeValue(… in set_split_dim()
|
| D | split_v.cc | 34 auto split_dim = GetValue<int64_t>(primitive->GetAttr("split_dim")); in InferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | split_infer.c | 46 …int split_dim = param->split_dim_ < 0 ? ((int)(input->shape_size_)) + param->split_dim_ : param->s… in SplitInferShape() local
|
| D | split_with_over_lap_infer.c | 34 int split_dim = param->split_dim_; in SplitWithOverlapInferShape() local
|
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | construct_operator_test.cc | 83 int64_t split_dim = args[1]; in TEST_F() local
131 int64_t split_dim = 0; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | split_base.c | 29 const int split_dim = split_param->split_dim_; in DoSplit() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | split_int8.c | 30 const int split_dim = param->split_dim_; in Int8DoSplit() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | split_base.cc | 47 auto split_dim = param->split_dim_; in CheckAndInitSplitParam() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 95 int64_t split_dim = args[TRANSFER_PERMUTE_SPLIT_DIM_INDEX]; in StridedSliceOP() local
202 int64_t split_dim = args[TRANSFER_PERMUTE_SPLIT_DIM_INDEX]; in AlltoAllOP() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | activation_gpu_kernel.h | 89 const int split_dim = 4; in Init() local
|
| D | activation_grad_kernel.h | 96 const int split_dim = 4; in Init() local
|
| D | l2normalize_gpu_kernel.h | 217 const int split_dim = 4; in InferInAndOutDesc() local
|
| D | l2normalize_grad_gpu_kernel.h | 274 constexpr int split_dim = 4; in InferInAndOutDesc() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_n_optimizer.cc | 165 int64_t split_dim = -1; in NeedSkip() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | depthwise_conv2d_info.cc | 54 … int64_t split_dim, std::vector<tensor::TensorPtr> *split_constant_tensors) { in CreateSplitConstantTensors()
418 int64_t split_dim = kAxisCOut; in CreateConstantOutputsOfSplit() local
|
| D | conv2d_info.cc | 145 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, size_t split_num, in CreateOutputsOfSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 61 int64_t split_dim = AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitDim); in CreateSplitNode() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | array_reduce_gpu_kernel.h | 220 const int split_dim = 4; in InferInAndOutDesc() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | update_thor_gradient.h | 248 size_t split_dim; variable
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | thor.py | 76 def caculate_matmul_shape(matrix_A_dim, matrix_G_dim, split_dim): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _thor_ops.py | 689 def __init__(self, split_dim=1): argument
711 def __init__(self, split_dim=0): argument
|
| D | comm_ops.py | 681 def __init__(self, split_count, split_dim, concat_dim, group=GlobalComm.WORLD_COMM_GROUP): argument
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 153 def caculate_matmul_shape(matrix_a_dim, matrix_g_dim, split_dim): argument
|