Searched refs:batch_matrix_pow (Results 1 – 2 of 2) sorted by relevance
185 return batch_matrix_pow(matrix_tiled, powers)188 def batch_matrix_pow(matrices, powers): function
85 math_utils.batch_matrix_pow(batch, powers).eval(),