Searched refs:TopKCompare (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | ordering_op-inl.h | 253 MSHADOW_XINLINE bool TopKCompare(DType val1, index_t ind1, DType val2, index_t ind2, in TopKCompare() function
266 if (TopKCompare(val1[i1], ind1[i1], val2[i2], ind2[i2], is_ascend)) { in MergeTopK()
274 if (i2 < 0 || i1 >= 0 && TopKCompare(val2[i2], ind2[i2], val1[i1], ind1[i1], is_ascend)) { in MergeTopK()
308 for (index_t j = K; j-- && TopKCompare(cur_val, cur_ind, val_buff[j], in PartialSortSmallK()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | ordering_op-inl.h | 253 MSHADOW_XINLINE bool TopKCompare(DType val1, index_t ind1, DType val2, index_t ind2, in TopKCompare() function
266 if (TopKCompare(val1[i1], ind1[i1], val2[i2], ind2[i2], is_ascend)) { in MergeTopK()
274 if (i2 < 0 || i1 >= 0 && TopKCompare(val2[i2], ind2[i2], val1[i1], ind1[i1], is_ascend)) { in MergeTopK()
308 for (index_t j = K; j-- && TopKCompare(cur_val, cur_ind, val_buff[j], in PartialSortSmallK()
|