def _get_value_shape(self): if self._shape_fully_defined: return tf.TensorShape([5]) return tf.TensorShape(None)