Searched refs:kAxisCIn (Results 1 – 3 of 3) sorted by relevance
53 if (is_any_not_none(strategys[0][kAxisCIn])) { in CheckStrategy()56 splits_ = strategys[0][kAxisCIn]; in CheckStrategy()57 if (strategys[0][kAxisCIn] != strategys[1][kAxisCIn]) { in CheckStrategy()274 …int cin_strategy_sum = std::accumulate(strategys[0][kAxisCIn].begin(), strategys[0][kAxisCIn].end(… in ConstructOutputCNodes()315 MS_CHECK_INT_MUL_NOT_OVERFLOW(in_channel, strategys[0][kAxisCIn][0], RET_ERROR); in ConstructOutputCNodes()317 prim->set_in_channel(in_channel * strategys[0][kAxisCIn][0] / cin_strategy_sum); in ConstructOutputCNodes()319 … prim->set_in_channel(in_channel - (in_channel * strategys[0][kAxisCIn][0] / cin_strategy_sum)); in ConstructOutputCNodes()
45 constexpr int32_t kAxisCIn = 3; variable
143 if (is_any_not_none(strategys[0][kAxisCIn])) { in CheckStrategy()145 splits_ = strategys[0][kAxisCIn]; in CheckStrategy()147 split_dim_ = kAxisCIn; in CheckStrategy()