Home
last modified time | relevance | path

Searched defs:num_split (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dsplit_fission.cc42 int64_t num_split) { in SetAttrForSplitVNode()
48 size_t GetSmallSplitSize(const AnfNodePtr &split_node, int64_t split_dim, int64_t num_split) { in GetSmallSplitSize()
81 …peAndTypeId(const CNodePtr &origin_cnode, int64_t split_dim, int64_t split_size, int64_t num_split, in CreateOutputShapeAndTypeId()
101 int64_t num_split) { in SetAttrAndAbstractForBaseSplitv()
124 …r SplitFission::DoFission(const FuncGraphPtr &func_graph, const CNodePtr &cnode, int64_t num_split, in DoFission()
203 auto num_split = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrOutputNum); in Process() local
Dsplitv_fission.cc38 auto num_split = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrNumSplit); in Process() local
/third_party/mindspore/mindspore/core/ops/
Dsplit_v.cc40 auto num_split = GetValue<int64_t>(primitive->GetAttr("num_split")); in InferShape() local
85 auto num_split = GetValue<int64_t>(prim->GetAttr("num_split")); in InferType() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dsplit_base.c26 const int num_split = split_param->num_split_; in DoSplit() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dsplit_int8.c27 const int num_split = param->num_split_; in Int8DoSplit() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dsplit.cc111 size_t num_split = UP_DIV(shape_dim, param->split_sizes_[0]); in AlignSplitSizes() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Ddepthwise_conv2d_info.cc38 void SplitConstantData(char *in_data, char **out_data, int64_t num_split, int64_t split_dim_size, i… in SplitConstantData()
/third_party/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_pack.c650 int num_split = src_type.width * src_type.length / 128; in lp_build_pack2() local
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py6504 def __init__(self, size_splits, split_dim, num_split): argument