def call(self, inputs, training=None, mask=None):
    assert isinstance(inputs, tf.Tensor)
    assert (inputs.dtype in [tf.uint8, tf.int64])
    inputs = (tf.cast(inputs, tf.float32) / 255)
    inputs_encoded = self.encoder_obj(inputs, training=training, mask=mask)
    return inputs_encoded
def call(self, inputs, training=None, mask=None):
    assert isinstance(inputs, tf.Tensor)
    assert (inputs.dtype in [tf.uint8, tf.int64])
    inputs = (tf.cast(inputs, tf.float32) / 255)
    inputs_encoded = self.encoder_obj(inputs, training=training, mask=mask)
    return inputs_encoded