Searched defs:mat2_size0 (Results 1 – 2 of 2) sorted by relevance
312 &mat2_size0 = mat2_shape[0], &mat2_size1 = mat2_shape[1]; in bmm_nested_cuda() local358 const int64_t &mat2_size0 = mat2_shape[0]; in bmm_nested_cuda() local
46 & mat2_size0 = mat2_shape[0], & mat2_size1 = mat2_shape[1]; in bmm_nested() local