code: def test_unknown_size_inputs() -> None:
      """
      Test for #725 and #734. When the shape of the Gaussian's mean had at least
      one unknown parameter, `gauss_kl` would blow up. This happened because
      `tf.size` can only output types `tf.int32` or `tf.int64`.
      """
      mu: AnyNDArray = np.ones([1, 4], dtype=default_float())
      sqrt: AnyNDArray = np.ones([4, 1, 1], dtype=default_float())
  
      known_shape = gauss_kl(*map(tf.constant, [mu, sqrt]))
      unknown_shape = gauss_kl(mu, sqrt)
  
      np.testing.assert_allclose(known_shape, unknown_shape)
  
imports: from unittest.mock import patch
  import numpy
  import pytest
  import tensorflow
  from check_shapes import ShapeChecker, check_shapes
  from numpy.testing import assert_allclose, assert_almost_equal
  import gpflow
  from gpflow import Parameter, default_float, default_jitter
  from gpflow.base import AnyNDArray, TensorType
  from gpflow.inducing_variables import InducingPoints
  from gpflow.kernels import Kernel
  from gpflow.kullback_leiblers import gauss_kl, prior_kl
  from gpflow.utilities.bijectors import triangular
label_desc: Use image node_id 13 for calling a global function with example usage: test_unknown_size_inputs() without return types