Searched defs:m_first_row (Results 1 – 2 of 2) sorted by relevance
247 int64_t m_first_row, in tril_indices_kernel()285 auto m_first_row = offset > 0 ? in tril_indices_cuda() local323 int64_t m_first_row, in triu_indices_kernel()362 auto m_first_row = offset > 0 ? in triu_indices_cuda() local
44 auto m_first_row = offset > 0 ? in get_tril_size() local