Searched refs:tpu_config (Results 1 – 12 of 12) sorted by relevance
28 from tensorflow.python.tpu import tpu_config215 use_tpu and config.tpu_config.num_cores_per_replica)217 num_cores_per_replica = config.tpu_config.num_cores_per_replica294 str(self._config.tpu_config.num_cores_per_replica))335 return self._config.tpu_config.input_partition_dims365 num_cores_per_replica = self._config.tpu_config.num_cores_per_replica398 (self._config.tpu_config.per_host_input_for_training is399 tpu_config.InputPipelineConfig.PER_SHARD_V1))403 return (self._config.tpu_config.per_host_input_for_training is404 tpu_config.InputPipelineConfig.PER_HOST_V2)[all …]
192 tpu_config=None, argument212 self._tpu_config = tpu_config or TPUConfig()268 def tpu_config(self): member in RunConfig279 tpu_config = kwargs.pop('tpu_config')281 new_instance._tpu_config = tpu_config # pylint: disable=protected-access
26 from tensorflow.python.tpu import tpu_config as tpu_config_lib80 tpu_config=tpu_config_lib.TPUConfig(num_shards=0))85 tpu_config=tpu_config_lib.TPUConfig(iterations_per_loop=0))171 self.assertIsNone(config.tpu_config.tpu_job_name)177 self.assertEqual('_my_new_name', config.tpu_config.tpu_job_name)
68 from tensorflow.python.tpu import tpu_config461 ctx.config.tpu_config.initial_infeed_sleep_secs)1005 if (ctx.config.tpu_config.eval_training_input_configuration is1006 tpu_config.InputPipelineConfig.SLICED):1010 if (ctx.config.tpu_config.eval_training_input_configuration is1011 tpu_config.InputPipelineConfig.SLICED):1910 if (self._ctx.config.tpu_config.per_host_input_for_training is1911 tpu_config.InputPipelineConfig.BROADCAST):2305 if config is None or not isinstance(config, tpu_config.RunConfig):2320 if (config.tpu_config.per_host_input_for_training is[all …]
63 "tpu_config.py",
14 from tensorflow.contrib.tpu.python.tpu import tpu_config32 run_config = tpu_config.RunConfig(
77 from tensorflow.contrib.tpu.python.tpu.tpu_config import *
102 run_config = tpu_config.RunConfig(107 tpu_config=tpu_config.TPUConfig(FLAGS.iterations, FLAGS.num_shards),)
50 "python/tpu/tpu_config.py",
31 from tensorflow.contrib.tpu.python.tpu import tpu_config146 self._config = tpu_config.RunConfig(model_dir=self._model_dir)250 self._config = tpu_config.RunConfig(model_dir=self._model_dir)289 config=tpu_config.RunConfig(
22 from tensorflow.python.tpu.tpu_config import *
202 tpu_config=tf.contrib.tpu.TPUConfig(