Searched refs:kAxisH (Results 1 – 5 of 5) sorted by relevance
69 if (is_any_not_none(strategys[0][kAxisH])) { in CheckStrategy()72 splits_ = strategys[0][kAxisH]; in CheckStrategy()78 if (is_any_not_none(strategys[1][kAxisH])) { in CheckStrategy()246 …auto feature_split_cnode = CreateOutputsOfSplit(cnode_, 0, &feature_split_outputs, kAxisH, dev_num… in InferParallelCNodes()380 concat_dim = kAxisH; in InferReplaceOp()
157 if (is_any_not_none(strategys[0][kAxisH])) { in CheckStrategy()159 splits_ = strategys[0][kAxisH]; in CheckStrategy()161 split_dim_ = kAxisH; in CheckStrategy()167 if (is_any_not_none(strategys[1][kAxisH])) { in CheckStrategy()346 int64_t input_h = input_shape.at(kAxisH); in CreateOutputsOfSplit()
47 constexpr int32_t kAxisH = 1; variable
281 int64_t input_h = input_shape.at(kAxisH); in AdJustConvPrim()
130 int64_t input_h = in_shape.at(kAxisH); in CalSplitInShape()