Searched defs:total_cols (Results 1 – 2 of 2) sorted by relevance
39 int total_rows, int total_cols, T* __restrict__ output) { in concat_fixed_kernel()64 IntType total_cols, T* output) { in concat_variable_kernel()
114 IntType total_rows, IntType total_cols, in split_v_kernel()211 int total_rows, int total_cols, in Run()