| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| D | functional_rnn.py | 282 use_tpu=False, argument
306 use_tpu=use_tpu,
330 use_tpu=False, argument
394 time_major=time_major, scope=fw_scope, use_tpu=use_tpu)
410 assert not use_tpu, (
429 use_tpu=use_tpu,
|
| D | recurrent.py | 285 use_tpu, argument
332 compiled = use_tpu
334 dev_t_type = dtypes.int32 if use_tpu else dtypes.int64
432 dev_t = math_ops.cast(t, dtypes.int32) if use_tpu else math_ops.cast(
572 dev_t = math_ops.cast(t, dtypes.int32) if use_tpu else math_ops.cast(
661 use_tpu=False, argument
740 use_tpu=use_tpu,
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 100 use_tpu=True, argument
217 if use_tpu:
235 is_on_tpu = _is_on_tpu(mode, use_tpu, eval_on_tpu)
254 use_tpu=use_tpu,
264 def _is_on_tpu(mode, use_tpu, eval_on_tpu): argument
266 return use_tpu
|
| D | tpu_gan_estimator_test.py | 123 is_on_tpu=FLAGS.use_tpu,
176 use_tpu=FLAGS.use_tpu,
273 use_tpu=FLAGS.use_tpu,
292 use_tpu=FLAGS.use_tpu,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_context.py | 201 use_tpu, argument
208 self._use_tpu = use_tpu
210 if not use_tpu and eval_on_tpu:
215 use_tpu and config.tpu_config.num_cores_per_replica)
706 predict_batch_size, use_tpu): argument
710 predict_batch_size, use_tpu)
732 predict_batch_size, use_tpu, eval_on_tpu, argument
745 predict_batch_size, use_tpu)
748 predict_batch_size, use_tpu, eval_on_tpu,
|
| D | tpu.py | 666 use_tpu=True, argument
836 num_replicas=num_replicas, use_tpu=use_tpu, **metadata_kwargs)
935 if use_tpu:
|
| D | tpu_estimator.py | 2224 use_tpu=True, argument
2313 if use_tpu:
2362 use_tpu, eval_on_tpu, experimental_embedding_config_spec)
3795 use_tpu=True,
|
| /external/tensorflow/tensorflow/contrib/tpu/ |
| D | README.md | 38 use_tpu=FLAGS.use_tpu,
|
| D | tpu_estimator.md | 111 use_tpu=FLAGS.use_tpu,
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | main_estimator_tpu.py | 137 if FLAGS.use_tpu:
190 if FLAGS.use_tpu:
221 use_tpu=FLAGS.use_tpu,
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | pruning.py | 218 use_tpu=False)
329 self._spec.name, use_resource=self._spec.use_tpu) as scope:
458 if not self._spec.use_tpu:
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TPUReplicateMetadata.pbtxt | 23 name: "use_tpu"
|
| D | api_def_TPUReplicate.pbtxt | 58 name: "use_tpu"
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla_test.py | 303 hparam.HParams(use_tpu=False)),
327 hparam.HParams(use_tpu=True)),
|
| /external/tensorflow/tensorflow/contrib/model_pruning/ |
| D | README.md | 64 | use_tpu | bool | False | Training using TPUs? |
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3849 …replicas\', \'output_types\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_assig…
3853 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3849 …replicas\', \'output_types\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_assig…
3853 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 82776 name: "use_tpu"
82881 name: "use_tpu"
82967 name: "use_tpu"
83028 name: "use_tpu"
|
| D | ops_history.v2.pbtxt | 82149 name: "use_tpu"
82254 name: "use_tpu"
82340 name: "use_tpu"
82401 name: "use_tpu"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 39931 name: "use_tpu"
40017 name: "use_tpu"
|