Searched defs:stride_an (Results 1 – 1 of 1) sorted by relevance
1645 def reduce_sum_kernel(a_ptr, c_ptr, stride_am, stride_an): argument1686 def argmax_kernel(a_ptr, c_ptr, stride_am, stride_an): argument