Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/_api/v2/experimental/dtensor/__init__.py: 100%
43 statements
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
1# This file is MACHINE GENERATED! Do not edit.
2# Generated by: tensorflow/python/tools/api/generator/create_python_api.py script.
3"""Public API for tf.experimental.dtensor namespace.
4"""
6import sys as _sys
8from tensorflow.dtensor.python.accelerator_util import initialize_accelerator_system
9from tensorflow.dtensor.python.accelerator_util import initialize_accelerator_system as initialize_multi_client
10from tensorflow.dtensor.python.accelerator_util import initialize_accelerator_system as initialize_tpu_system
11from tensorflow.dtensor.python.accelerator_util import shutdown_accelerator_system
12from tensorflow.dtensor.python.accelerator_util import shutdown_accelerator_system as shutdown_tpu_system
13from tensorflow.dtensor.python.api import call_with_layout
14from tensorflow.dtensor.python.api import check_layout
15from tensorflow.dtensor.python.api import copy_to_mesh
16from tensorflow.dtensor.python.api import default_mesh
17from tensorflow.dtensor.python.api import device_name
18from tensorflow.dtensor.python.api import fetch_layout
19from tensorflow.dtensor.python.api import is_dtensor
20from tensorflow.dtensor.python.api import pack
21from tensorflow.dtensor.python.api import relayout
22from tensorflow.dtensor.python.api import run_on
23from tensorflow.dtensor.python.api import unpack
24from tensorflow.dtensor.python.config import client_id
25from tensorflow.dtensor.python.config import full_job_name
26from tensorflow.dtensor.python.config import heartbeat_enabled
27from tensorflow.dtensor.python.config import job_name
28from tensorflow.dtensor.python.config import jobs
29from tensorflow.dtensor.python.config import local_devices
30from tensorflow.dtensor.python.config import num_clients
31from tensorflow.dtensor.python.config import num_global_devices
32from tensorflow.dtensor.python.config import num_local_devices
33from tensorflow.dtensor.python.config import preferred_device_type
34from tensorflow.dtensor.python.d_checkpoint import DTensorCheckpoint
35from tensorflow.dtensor.python.d_variable import DVariable
36from tensorflow.dtensor.python.input_util import DTensorDataset
37from tensorflow.dtensor.python.layout import Layout
38from tensorflow.dtensor.python.layout import MATCH
39from tensorflow.dtensor.python.layout import Mesh
40from tensorflow.dtensor.python.layout import UNSHARDED
41from tensorflow.dtensor.python.mesh_util import barrier
42from tensorflow.dtensor.python.mesh_util import create_distributed_mesh
43from tensorflow.dtensor.python.mesh_util import create_mesh
44from tensorflow.dtensor.python.save_restore import enable_save_as_bf16
45from tensorflow.dtensor.python.save_restore import name_based_restore
46from tensorflow.dtensor.python.save_restore import name_based_save
47from tensorflow.dtensor.python.save_restore import sharded_save
48from tensorflow.dtensor.python.tpu_util import create_tpu_mesh