LostTech.TensorFlow : API Documentation

Type I_InternalTPUContext

Namespace tensorflow_estimator.python.estimator.tpu.tpu_context

Interfaces IPythonObjectContainer

Methods

Properties

Public instance methods

object is_running_on_cpu(object is_export_mode)

object with_mode(object mode)

Public properties

object batch_size_for_input_fn get;

object batch_size_for_model_fn get;

object config get;

object device_assignment get;

object embedding_config get;

object global_batch_size get;

object input_partition_dims get;

object master_job get;

object mode get;

object model_parallelism_enabled get;

object num_cores get;

object num_hosts get;

object num_of_cores_per_host get;

object num_of_replicas_per_host get;

object num_replicas get;

object tpu_device_placement_function get;

object tpu_host_placement_function get;