Searched defs:num_split (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | split_fission.cc | 42 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
|
| D | splitv_fission.cc | 38 auto num_split = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrNumSplit); in Process() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | split_v.cc | 40 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/ |
| D | split_base.c | 26 const int num_split = split_param->num_split_; in DoSplit() local
|
| /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
|
| /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
|
| /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()
|
| /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
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 6504 def __init__(self, size_splits, split_dim, num_split): argument
|