Searched defs:mat2_shape (Results 1 – 2 of 2) sorted by relevance
310 …const IntArrayRef &self_shape = get_size_for_index(self, i), &mat2_shape = get_size_for_index(mat2… in bmm_nested_cuda() local355 const IntArrayRef& mat2_shape = get_size_for_index(mat2, i); in bmm_nested_cuda() local
44 const IntArrayRef& mat2_shape = get_size_for_index(mat2, i); in bmm_nested() local