Searched refs:batch_matrices (Results 1 – 1 of 1) sorted by relevance
125 def broadcast_matrix_batch_dims(batch_matrices, name=None): argument173 name or "broadcast_matrix_batch_dims", values=batch_matrices):174 check_ops.assert_proper_iterable(batch_matrices)175 batch_matrices = list(batch_matrices)177 for i, mat in enumerate(batch_matrices):178 batch_matrices[i] = ops.convert_to_tensor(mat)179 assert_is_batch_matrix(batch_matrices[i])181 if len(batch_matrices) < 2:182 return batch_matrices190 bcast_batch_shape = batch_matrices[0].get_shape()[:-2][all …]