Searched defs:reduction_dimension (Results 1 – 2 of 2) sorted by relevance
1375 int64_t reduction_dimension; in ApproxTopKShape() local
5797 reduction_dimension=-1, argument5860 reduction_dimension=-1, argument