Searched refs:rank_after_pack (Results 1 – 1 of 1) sorted by relevance
34 Status GetAxisForPackAndUnpack(InferenceContext* c, int32 rank_after_pack, in GetAxisForPackAndUnpack() argument37 if (*axis < -1 * rank_after_pack || *axis >= rank_after_pack) { in GetAxisForPackAndUnpack()39 -1 * rank_after_pack, ",", rank_after_pack, in GetAxisForPackAndUnpack()42 if (*axis < 0) *axis = (rank_after_pack + *axis); in GetAxisForPackAndUnpack()