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