| #!/usr/bin/env vpython3 |
| # Copyright 2022 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| # Credits to the The Fuchsia Authors for creating this file. |
| """This script is used to fetch the tensorflow 2.7.0 pip package (via vpython) |
| for use by MLGO during LLVM compile. The vpython spec is hand-created, with the |
| help of pip. The transitive dependencies can be retrieved by using pipdeptree, |
| a pip package, by running `pipdeptree -p tensorflow`. |
| """ |
| # [VPYTHON:BEGIN] |
| # python_version: "3.8" |
| # wheel: < |
| # name: "infra/python/wheels/absl-py-py3" |
| # version: "version:0.11.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/astunparse-py2_py3" |
| # version: "version:1.6.3" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/cachetools-py3" |
| # version: "version:4.2.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/certifi-py2_py3" |
| # version: "version:2021.5.30" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/charset_normalizer-py3" |
| # version: "version:2.0.4" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/flatbuffers-py3" |
| # version: "version:2.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/gast-py3" |
| # version: "version:0.4.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/google-auth-oauthlib-py3" |
| # version: "version:0.4.5" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/google-auth-py2_py3" |
| # version: "version:1.35.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/google-pasta-py3" |
| # version: "version:0.2.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/grpcio/${vpython_platform}" |
| # version: "version:1.39.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/h5py/${vpython_platform}" |
| # version: "version:3.6.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/idna-py3" |
| # version: "version:3.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/keras-py3" |
| # version: "version:2.7.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/keras-preprocessing-py2_py3" |
| # version: "version:1.1.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/libclang/${vpython_platform}" |
| # version: "version:12.0.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/markdown-py3" |
| # version: "version:3.3.4" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/numpy/${vpython_platform}" |
| # version: "version:1.22.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/oauthlib-py2_py3" |
| # version: "version:3.1.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/opt-einsum-py3" |
| # version: "version:3.3.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/protobuf-py2_py3" |
| # version: "version:3.17.3" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/pyasn1-py2_py3" |
| # version: "version:0.4.8" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/pyasn1_modules-py2_py3" |
| # version: "version:0.2.8" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/requests-py2_py3" |
| # version: "version:2.26.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/requests-oauthlib-py2_py3" |
| # version: "version:1.3.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/rsa-py3" |
| # version: "version:4.7.2" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/setuptools-py3" |
| # version: "version:57.4.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/six-py2_py3" |
| # version: "version:1.16.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorboard-py3" |
| # version: "version:2.7.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorboard-data-server-py3" |
| # version: "version:0.6.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorboard-plugin-wit-py3" |
| # version: "version:1.8.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorflow/${vpython_platform}" |
| # version: "version:2.7.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorflow-estimator-py3" |
| # version: "version:2.7.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/tensorflow-io-gcs-filesystem/${vpython_platform}" |
| # version: "version:0.23.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/termcolor-py2_py3" |
| # version: "version:1.1.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/typing-extensions-py3" |
| # version: "version:4.0.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/urllib3-py2_py3" |
| # version: "version:1.26.6" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/werkzeug-py3" |
| # version: "version:2.0.1" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/wheel-py2_py3" |
| # version: "version:0.37.0" |
| # > |
| # wheel: < |
| # name: "infra/python/wheels/wrapt/${vpython_platform}" |
| # version: "version:1.13.3" |
| # > |
| # [VPYTHON:END] |
| import importlib |
| import os |
| |
| spec = importlib.util.find_spec("tensorflow") |
| print(os.path.dirname(spec.origin)) |
