Home
last modified time | relevance | path

Searched refs:kAxisW (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.h48 constexpr int32_t kAxisW = 2; variable
Ddepthwise_conv2d_info.cc163 if (is_any_not_none(strategys[0][kAxisW])) { in CheckStrategy()
171 if (is_any_not_none(strategys[1][kAxisW])) { in CheckStrategy()
347 int64_t input_w = input_shape.at(kAxisW); in CreateOutputsOfSplit()
Dconv2d_info.cc74 if (is_any_not_none(strategys[0][kAxisW])) { in CheckStrategy()
82 if (is_any_not_none(strategys[1][kAxisW])) { in CheckStrategy()
Dmulti_conv_info.cc282 int64_t input_w = input_shape.at(kAxisW); in AdJustConvPrim()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc126 if (in_shape.size() < kAxisW) { in CalSplitInShape()
131 int64_t input_w = in_shape.at(kAxisW); in CalSplitInShape()