def run():
package_dir = common_setup.get_package_dir(prefix=("integrations",
"tensorflow",
"bindings", "python"))
packages = setuptools.find_namespace_packages(package_dir,
include=[
"pyiree.tf.compiler",
"pyiree.tf.compiler.*",
"pyiree.tf.support",
"pyiree.tf.support.*"
],
exclude=["*.CMakeFiles"])
print("Found packages:", packages)
if not packages:
print("ERROR: Did not find packages under", package_dir)
sys.exit(1)
setup_kwargs = common_setup.get_setup_defaults(
sub_project="tf",
description="IREE TensorFlow Compiler",
package_dir=package_dir)
common_setup.setup(packages=packages,
ext_modules=[
setuptools.Extension(
name="pyiree.tf.compiler.binding", sources=[]),
],
**setup_kwargs)
def run():
packages = setuptools.find_namespace_packages(
common_setup.get_package_dir(),
include=["pyiree.compiler", "pyiree.compiler.*"],
exclude=["*.CMakeFiles"])
print("Found packages:", packages)
setup_kwargs = common_setup.get_setup_defaults(
sub_project="compiler", description="IREE Generic Compiler")
common_setup.setup(packages=packages, **setup_kwargs)
def run():
packages = setuptools.find_namespace_packages(
common_setup.get_package_dir(),
include=["pyiree.rt", "pyiree.rt.*"],
exclude=["*.CMakeFiles"])
print("Found packages:", packages)
setup_kwargs = common_setup.get_setup_defaults(
sub_project="rt",
description="IREE Runtime Components (for executing compiled programs)"
)
common_setup.setup(packages=packages, **setup_kwargs)
def run():
package_dir = find_bazel_runfiles_dir()
packages = setuptools.find_namespace_packages(package_dir,
include=[
"pyiree.tf.compiler",
"pyiree.tf.compiler.*",
"pyiree.tf.support",
"pyiree.tf.support.*"
],
exclude=["*.CMakeFiles"])
print("Found packages:", packages)
if not packages:
print("ERROR: Did not find packages under", package_dir)
sys.exit(1)
setup_kwargs = common_setup.get_setup_defaults(
sub_project="tf",
description="IREE TensorFlow Compiler",
package_dir=package_dir)
common_setup.setup(packages=packages, **setup_kwargs)
def main():
u'''[SETUP] common variables/datasets'''
cv = cs.setup()
locals().update(cv)
u'''[PREPARE] trainer'''
DCA = aat.trainer(cv)
u'''[TRAIN] neural networks'''
DCA.train()
u'''[GENERATE] samples'''
DCA.generate()
u'''[TEST] discrimination'''
DCA.discriminate()
u'''[STAT] spanning prob distribution'''
DCA.compare()