| 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() | |