Searched refs:GetAxisForPackAndUnpack (Results 1 – 1 of 1) sorted by relevance
34 Status GetAxisForPackAndUnpack(InferenceContext* c, int32 rank_after_pack, in GetAxisForPackAndUnpack() function339 TF_RETURN_IF_ERROR(GetAxisForPackAndUnpack(c, rank + 1, &axis)); in __anondb9326b20302()422 TF_RETURN_IF_ERROR(GetAxisForPackAndUnpack(c, rank, &axis)); in __anondb9326b20502()