python_version = "PY3",
    tags = [
        "no_cuda_asan",  # Not needed, and there were issues with timeouts.
        "no_oss",  # Avoid downloading mnist data set in oss.
        "nomultivm",  # Not needed. Save some resources and test time.
        "notap",  # The test is too long to run as part of llvm presubmits (b/173661843).
        "notpu",  # Takes too long (b/192305423)