Searched refs:transform_matrices (Results 1 – 1 of 1) sorted by relevance
366 def matrices_to_flat_transforms(transform_matrices): argument386 transform_matrices = ops.convert_to_tensor(387 transform_matrices, name="transform_matrices")388 if transform_matrices.shape.ndims not in (2, 3):390 "Matrices should be 2D or 3D, got: %s" % transform_matrices)392 transforms = array_ops.reshape(transform_matrices,