code: def _square_size(dense_shape):
      dense_shape = tf.unstack(dense_shape)
      size = dense_shape[0]
      for d in dense_shape[1:]:
          tf.debugging.assert_equal(size, d)
      return size
  
file_path: spektral/spektral/layers/ops/sparse.py
imports: import tensorflow
  from tensorflow.python.ops import gen_sparse_ops
  from .None import ops
label_desc: Use image node_id 3 for calling a global function with example usage: _square_size(dense_shape) and returns: size