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