import tensorflow as tf #from e2eflow.test.test_image_warp import ImageWarpTest #from e2eflow.test.test_losses import LossesTest #from e2eflow.test.ops.backward_warp import BackwardWarpTest #from e2eflow.test.ops.forward_warp import ForwardWarpTest #from e2eflow.test.ops.downsample import DownsampleTest from e2eflow.test.ops.correlation import CorrelationTest if __name__ == "__main__": tf.test.main()