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

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""" 

5 

6import sys as _sys 

7 

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