Searched refs:tensor_end (Results 1 – 2 of 2) sorted by relevance
718 size_t tensor_end = in.size(); in topK() local722 while (in_p < tensor_end) { in topK()
4635 size_t tensor_end = in.size(); in fwdTopK() local4639 while (in_p < tensor_end) { in fwdTopK()