Home
last modified time | relevance | path

Searched defs:split_dim_size (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dsplit_with_over_lap_infer.c50 int split_dim_size = input_shape[split_dim]; in SplitWithOverlapInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dsplit_with_over_lap_base.cc40 auto split_dim_size = shape[param_->split_dim_]; in CalculateSplitedShapes() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Ddepthwise_conv2d_info.cc43 void SplitConstantData(char *in_data, char **out_data, int64_t num_split, int64_t split_dim_size, i… in SplitConstantData()
72 int64_t split_dim_size = constant_shape[split_dim]; in CreateSplitConstantTensors() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc439 …UpdateRatioWithPadStride(int64_t *ratio, size_t ratio_len, size_t split_size, int split_dim_size) { in UpdateRatioWithPadStride()