Adding missing modules to __init__.py files for documentation generat…

…ion.

PiperOrigin-RevId: 354625490

tensorflow_graphics/geometry/representation/mesh/__init__.py

@@ -17,6 +17,7 @@
 from __future__ import print_function
 
 from tensorflow_graphics.geometry.representation.mesh import normals
+from tensorflow_graphics.geometry.representation.mesh import sampler
 from tensorflow_graphics.geometry.representation.mesh import utils
 from tensorflow_graphics.util import export_api as _export_api
 

tensorflow_graphics/geometry/representation/mesh/sampler.py

@@ -41,6 +41,7 @@
 from tensorflow_graphics.geometry.representation import triangle
 from tensorflow_graphics.geometry.representation.mesh import normals
 from tensorflow_graphics.util import asserts
+from tensorflow_graphics.util import export_api
 from tensorflow_graphics.util import shape
 
 
@@ -395,3 +396,6 @@ def area_weighted_random_sample_triangle_mesh(vertex_attributes,
         face_weights=triangle_areas,
         seed=seed,
         stateless=stateless)
+
+# API contains all public functions and classes.
+__all__ = export_api.get_functions_and_classes()

tensorflow_graphics/geometry/transformation/__init__.py

@@ -17,6 +17,7 @@
 from __future__ import print_function
 
 from tensorflow_graphics.geometry.transformation import axis_angle
+from tensorflow_graphics.geometry.transformation import dual_quaternion
 from tensorflow_graphics.geometry.transformation import euler
 from tensorflow_graphics.geometry.transformation import linear_blend_skinning
 from tensorflow_graphics.geometry.transformation import look_at

tensorflow_graphics/geometry/transformation/dual_quaternion.py

@@ -37,6 +37,7 @@
 
 from tensorflow_graphics.geometry.transformation import quaternion
 from tensorflow_graphics.util import asserts
+from tensorflow_graphics.util import export_api
 from tensorflow_graphics.util import shape
 
 
@@ -74,3 +75,6 @@ def conjugate(dual_quaternion, name=None):
     return tf.concat((quaternion.conjugate(quaternion_real),
                       quaternion.conjugate(quaternion_dual)),
                      axis=-1)
+
+# API contains all public functions and classes.
+__all__ = export_api.get_functions_and_classes()

tensorflow_graphics/nn/layer/__init__.py

@@ -17,6 +17,7 @@
 from __future__ import print_function
 
 from tensorflow_graphics.nn.layer import graph_convolution
+from tensorflow_graphics.nn.layer import pointnet
 from tensorflow_graphics.util import export_api as _export_api
 
 # API contains submodules of tensorflow_graphics.geometry.

tensorflow_graphics/nn/layer/pointnet.py

@@ -39,6 +39,7 @@
 """
 
 import tensorflow as tf
+from tensorflow_graphics.util import export_api
 
 
 class PointNetConv2Layer(tf.keras.layers.Layer):
@@ -239,3 +240,7 @@ def loss(labels, logits):
     cross_entropy = tf.nn.sparse_softmax_cross_entropy_with_logits
     residual = cross_entropy(labels, logits)
     return tf.reduce_mean(input_tensor=residual)
+
+
+# API contains all public functions and classes.
+__all__ = export_api.get_functions_and_classes()

tensorflow_graphics/rendering/camera/__init__.py

@@ -18,6 +18,7 @@
 
 from tensorflow_graphics.rendering.camera import orthographic
 from tensorflow_graphics.rendering.camera import perspective
+from tensorflow_graphics.rendering.camera import quadratic_radial_distortion
 from tensorflow_graphics.util import export_api as _export_api
 
 # API contains submodules of tensorflow_graphics.rendering.camera.