| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| 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 cudaError_t MatrixSplit(const size_t size, const size_t split_dim, const size_t dim, T *input_addr,… in MatrixSplit()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_splitv.py | 38 def __init__(self, size_split, split_dim, num_split, strategy1=None, strategy2=None): argument
51 def __init__(self, size_split, split_dim, num_split, strategy1=None, strategy2=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | split_with_overlap.cc | 31 int64_t split_dim, int64_t stride, int64_t pad_top, bool trans_format) { in Init()
58 void SplitWithOverlap::set_split_dim(int64_t split_dim) { (void)this->AddAttr(kSplitDim, api::MakeV… in set_split_dim()
|
| D | split_v.cc | 54 auto split_dim = GetValue<int64_t>(primitive->GetAttr("split_dim")); in SplitVInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | split_fission.cc | 28 …ForSplitVNode(const AnfNodePtr &splitv, const std::vector<int64_t> &size_splits, int64_t split_dim, in SetAttrForSplitVNode()
54 void CreateOutputShapeAndTypeId(const CNodePtr &origin_cnode, int64_t split_dim, in CreateOutputShapeAndTypeId()
77 … const std::vector<int64_t> &size_splits_base, int64_t split_dim, in SetAttrAndAbstractForBaseSplitv()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | glu_proto.cc | 29 int64_t split_dim; in IMPLEMT_COMMON_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | split_dynamic_fp32_coder.cc | 35 auto split_dim = param_->split_dim_; in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/ |
| D | split_mapper.cc | 46 auto split_dim = GetValue<int64_t>(axis_value); in Mapper() local
|
| /third_party/mindspore/mindspore-src/source/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-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | split_base.c | 29 const int split_dim = split_param->split_dim_; in DoSplit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | l2normalize_grad_gpu_kernel.h | 127 constexpr int split_dim = 4; in InferInAndOutDesc() local
|
| D | activation_grad_kernel.cc | 99 const int split_dim = 4; in Resize() local
|
| D | activation_gpu_kernel.cc | 113 const int split_dim = 4; in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | split_with_over_lap_infer.c | 34 int split_dim = param->split_dim_; in SplitWithOverlapInferShape() local
|
| D | split_infer.c | 100 …int split_dim = param->split_dim_ < 0 ? ((int)(input->shape_size_)) + param->split_dim_ : param->s… in SplitInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | split_with_size_op_pass.cc | 53 auto split_dim = split_dim_node->cast<ValueNodePtr>(); in SplitWithSizeMapperToSplitV() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/ |
| D | split_npu.cc | 40 auto split_dim = in_tensor.Shape().at(axis_); in Init() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | split_int8.c | 32 const int split_dim = param->split_dim_; in Int8DoSplit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | depthwise_conv2d_info.cc | 62 … int64_t split_dim, std::vector<tensor::TensorPtr> *split_constant_tensors) { in CreateSplitConstantTensors()
342 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, in CreateOutputsOfSplit()
429 int64_t split_dim = kAxisCOut; in CreateConstantOutputsOfSplit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | neighbor_exchange_v2_fusion.h | 30 int64_t split_dim = 0; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 138 int64_t split_dim = args[TRANSFER_PERMUTE_SPLIT_DIM_INDEX]; in ReplaceStridedSliceOpToSplitOp() local
187 int64_t split_dim = args[TRANSFER_PERMUTE_SPLIT_DIM_INDEX]; in StridedSliceOP() local
305 int64_t split_dim = args[TRANSFER_PERMUTE_SPLIT_DIM_INDEX]; in AlltoAllOP() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | split_base.cc | 47 auto split_dim = param->split_dim_; in CheckAndInitSplitParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | grouped_pairwise_exchange_alltoall.cc | 170 size_t split_dim = LongToSize(common::AnfAlgo::GetNodeAttr<int64_t>(alltoall, kAttrSplitDim)); in GetSplitDimFromAlltoall() local
179 CNodePtr NewSplitNode(const AnfNodePtr &input_node, size_t split_dim, size_t split_num) { in NewSplitNode()
200 CNodePtr NewSplitNode(const AnfNodePtr &input_node, size_t split_dim, size_t split_num, const Shape… in NewSplitNode()
334 void CreateNeighborExchangeNodes(const AnfNodePtr &input_node, size_t split_dim, size_t concat_dim, in CreateNeighborExchangeNodes()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/ |
| D | split_n_optimizer.cc | 169 int64_t split_dim = -1; in NeedSkip() local
|
| /third_party/mesa3d/src/gallium/drivers/etnaviv/ |
| D | etnaviv_ml_tp.c | 447 …unsigned split_dim = split_reshuffle(subgraph, operation, tp_core, tp_cores_used, in_dims, out_dim… in create_reshuffle_config() local
491 …unsigned split_dim = split_reshuffle(subgraph, operation, i, tp_cores_used, in_dims, out_dims, NUL… in create_reshuffle_config() local
|