Use tf.float32 as a dtype for tf.Tensors

sft/spline/surface_models.py

@@ -415,7 +415,7 @@ class SplineSphere(SplineSurface):
         self._south_pole = points[-1]
         self._coefs[2 * self.Mt:-2 * self.Mt] = points[1:-1]
         if tf.is_tensor(points):
-            self._coefs = tf.convert_to_tensor(self._coefs)
+            self._coefs = tf.convert_to_tensor(self._coefs, dtype=tf.float32)
 
         self._is_initialised = False
 
@@ -551,7 +551,7 @@ class SplineSphere(SplineSurface):
 
             # Support for GPU computations
             if tf.is_tensor(self._coefs):
-                phi = tf.convert_to_tensor(phi)
+                phi = tf.convert_to_tensor(phi, dtype=tf.float32)
 
             SplineSphere._phi[key] = phi
         return phi
@@ -658,7 +658,7 @@ class SplineSphere(SplineSurface):
 
         # GPU support
         if tf.is_tensor(self._coefs):
-            self._coefs = tf.convert_to_tensor(reshaped_coefs)
+            self._coefs = tf.convert_to_tensor(reshaped_coefs, dtype=tf.float32)
         else:
             self._coefs = reshaped_coefs
 
@@ -743,7 +743,7 @@ class SplineSphere(SplineSurface):
 
         # GPU support
         if tf.is_tensor(sample_points):
-            self._coefs = tf.convert_to_tensor(self._coefs)
+            self._coefs = tf.convert_to_tensor(self._coefs, dtype=tf.float32)
 
         vs0 = self._basis_s.value(0.0)
         vs1 = self._basis_s.value(1.0)