Searched refs:assert_is_batch_matrix (Results 1 – 1 of 1) sorted by relevance
248 def assert_is_batch_matrix(tensor): function327 assert_is_batch_matrix(batch_matrices[i])