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