Searched refs:batch_matrices (Results 1 – 1 of 1) sorted by relevance
273 def broadcast_matrix_batch_dims(batch_matrices, name=None): argument321 name or "broadcast_matrix_batch_dims", values=batch_matrices):322 check_ops.assert_proper_iterable(batch_matrices)323 batch_matrices = list(batch_matrices)325 for i, mat in enumerate(batch_matrices):326 batch_matrices[i] = ops.convert_to_tensor_v2_with_dispatch(mat)327 assert_is_batch_matrix(batch_matrices[i])329 if len(batch_matrices) < 2:330 return batch_matrices338 bcast_batch_shape = batch_matrices[0].shape[:-2][all …]