Searched refs:GetDimForAxis (Results 1 – 3 of 3) sorted by relevance
69 inline int64_t GetDimForAxis(int64_t axis, int64_t rank) { in GetDimForAxis() function
85 int64_t dim = GetDimForAxis(index.getSExtValue(), rank); in InferReductionOpType()
2824 axis = GetDimForAxis(axis, value_rank); in verify()