HomeSort by relevance Sort by last modified time
    Searched refs:tpu (Results 26 - 50 of 81) sorted by null

12 3 4

  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_function.py 22 from tensorflow.python.tpu.tpu_function import *
tpu_optimizer.py 22 from tensorflow.python.tpu.tpu_optimizer import *
tpu_sharding.py 22 from tensorflow.python.tpu.tpu_sharding import *
training_loop.py 22 from tensorflow.python.tpu.training_loop import *
util.py 22 from tensorflow.python.tpu.util import *
  /external/tensorflow/tensorflow/python/tpu/experimental/
__init__.py 15 """Experimental TPU library."""
22 from tensorflow.python.tpu import tpu_strategy_util
  /external/tensorflow/tensorflow/contrib/distribute/python/
tpu_strategy.py 15 """TPU Distribution Strategy.
26 from tensorflow.python.tpu.tpu_strategy_util import initialize_tpu_system
  /external/tensorflow/tensorflow/python/tpu/
functional.py 21 from tensorflow.python.tpu.ops import tpu_ops
tpu_strategy_util.py 15 """TPU specific APIs to be used in conjunction with TPU Strategy."""
30 from tensorflow.python.tpu import functional as tpu_functional_ops
31 from tensorflow.python.tpu import topology
32 from tensorflow.python.tpu import tpu
38 """Get the device spec for the first TPU host."""
41 [x for x in context.list_devices() if "device:TPU:" in x])
43 raise RuntimeError("Could not find any TPU devices")
56 @tf_export("tpu.experimental.initialize_tpu_system"
    [all...]
tpu_test.py 35 from tensorflow.python.tpu import tpu
36 from tensorflow.python.tpu import tpu_feed
37 from tensorflow.python.tpu import training_loop
44 """Test that control_flow_util can check that we're in a TPU context."""
47 context = tpu.TPUReplicateContext(b"context", 1, pivot=pivot)
84 tpu.rewrite(loop)
112 node.attr[tpu._TPU_REPLICATE_ATTR].s = b"0"
123 tpu.prune_unconnected_ops_from_xla(ops.get_default_graph())
127 tpu._TPU_REPLICATE_ATTR
    [all...]
topology_test.py 23 from tensorflow.python.tpu import topology
_tpu_estimator_embedding.py 15 """Tooling for support TPU embedding in TPUEstimator."""
26 from tensorflow.python.tpu import feature_column as tpu_fc
27 from tensorflow.python.tpu import tpu_embedding
28 from tensorflow.python.tpu.tpu_embedding import AdagradParameters
29 from tensorflow.python.tpu.tpu_embedding import AdamParameters
30 from tensorflow.python.tpu.tpu_embedding import StochasticGradientDescentParameters
182 # or TPU mode. So allow non-TPU embedding columns also.
tpu_optimizer.py 16 """Optimizer that implements cross-shard gradient reduction for TPU."""
26 from tensorflow.python.tpu import tpu_function
27 from tensorflow.python.tpu.ops import tpu_ops
32 """An optimizer that averages gradients across TPU shards."""
67 num_shards: The number of TPU shards.
bfloat16_test.py 26 from tensorflow.python.tpu import bfloat16
feature_column.py 15 """TPU Feature Column Library."""
27 from tensorflow.python.tpu import tpu
28 from tensorflow.python.tpu import tpu_function
47 """TPU embedding_column for `tf.feature_column.embedding_column`.
49 Note that the interface for TPU embedding_column is different from the non-TPU
50 version. The following args available for the non-TPU version are NOT
76 'categorical_column for tpu '
113 # that non-TPU Embedding column and non-TPU shared Embedding column handle th
    [all...]
  /external/tensorflow/tensorflow/python/distribute/cluster_resolver/
tpu_cluster_resolver.py 52 r'.*task:(?P<host_id>\d+)/.*device:TPU:(?P<core_id>\d+)$')
63 This is an implementation of cluster resolvers for the Google Cloud TPU
70 """Creates a new Cloud TPU API object.
78 A Google Cloud TPU API object.
89 'tpu', 'v1alpha1',
94 'tpu', 'v1alpha1',
121 total_cores: The total number of cores within the TPU system.
143 raise RuntimeError('TPU cores on each device is not the same. This '
181 tpu=None,
192 The ClusterResolver will then use the parameters to query the Cloud TPU API
    [all...]
tpu_cluster_resolver_test.py 139 TPUClusterResolver(tpu='')
158 'projects/test-project/locations/us-central1-c/nodes/test-tpu-1': {
168 tpu=['test-tpu-1'],
192 'projects/test-project/locations/us-central1-c/nodes/test-tpu-1': {
202 tpu=['test-tpu-1'],
219 'projects/test-project/locations/us-central1-c/nodes/test-tpu-1': {
229 tpu='test-tpu-1'
    [all...]
  /external/tensorflow/tensorflow/core/ops/
tpu_embedding_ops.cc 23 #include "tensorflow/core/protobuf/tpu/tpu_embedding_configuration.pb.h"
24 #include "tensorflow/core/tpu/tpu_embedding_optimization_parameters_utils.h"
25 #include "tensorflow/core/tpu/tpu_embedding_output_layout_utils.h"
31 // to CPUs. Embedding lookups on TPU systems are achieved by including the
35 // in the model, the size of the TPU system to be used, and the optimizer to
38 // 1. Pass this TPUEmbeddingConfiguration to tpu.initialize_system() as the
40 // 2. Use the LoadTPUEmbedding Ops to initialize the embedding tables in TPU
42 // 3. Use EnqueueTPUEmbeddingSparseBatch to provide the TPU with embedding
53 // TPU memories.
55 // TPU Embeddings use dedicated ops to enforce Host/TPU consistency in th
    [all...]
  /external/tensorflow/tensorflow/core/tpu/
tpu_embedding_optimization_parameters_utils.h 22 #include "tensorflow/core/protobuf/tpu/optimization_parameters.pb.h"
25 namespace tpu { namespace in namespace:tensorflow
70 // flushed to zero on the current TPU platforms and needs to continue to have
87 } // namespace tpu
tpu_embedding_output_layout_utils.cc 16 #include "tensorflow/core/tpu/tpu_embedding_output_layout_utils.h"
18 #include "tensorflow/core/protobuf/tpu/tpu_embedding_output_layout.pb.h"
21 namespace tpu { namespace in namespace:tensorflow
97 } // namespace tpu
  /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
main_estimator_tpu.py 15 """Cloud TPU Estimator workflow with RevNet train on ImageNet."""
44 https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/TPUEstimatorSpec
59 # TPU loop is finished, setting max_queue value to the same as number of
77 https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/TPUEstimatorSpec
102 """Model function required by the `tf.contrib.tpu.TPUEstimator` API.
111 An instance of `tf.contrib.tpu.TPUEstimatorSpec`
138 optimizer = tf.contrib.tpu.CrossShardOptimizer(optimizer)
156 return tf.contrib.tpu.TPUEstimatorSpec(
163 return tf.contrib.tpu.TPUEstimatorSpec(
173 return tf.contrib.tpu.TPUEstimatorSpec
    [all...]
  /external/tensorflow/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/
main.py 27 # Cloud TPU Cluster Resolvers
30 'Project name for the Cloud TPU-enabled project. If not specified, we '
35 help='GCE zone where the Cloud TPU is located in. If not specified, we '
38 'tpu', None, 'Name of the Cloud TPU for Cluster Resolvers. You must '
43 'service_addr', None, 'Address of TPU profiler service e.g. '
44 'localhost:8466, you must specify either this flag or --tpu.')
47 ' e.g. 10.0.1.2, 10.0.1.3. You can specify this flag with --tpu or '
48 '--service_addr to profile a subset of tpu nodes. You can also use only'
49 '--tpu and leave this flag unspecified to profile all the tpus.'
    [all...]
  /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
tpu_gan_estimator_test.py 15 """Tests for TF-GAN's TPU Estimator."""
31 from tensorflow.contrib.tpu.python.tpu import tpu_config
32 from tensorflow.contrib.tpu.python.tpu import tpu_estimator
33 from tensorflow.contrib.tpu.python.tpu import tpu_optimizer
52 flags.DEFINE_bool('use_tpu', False, 'Whether to run test on TPU or not.')
tpu_gan_estimator_impl.py 15 """A TF-GAN-backed GAN Estimator that works on TPU."""
24 from tensorflow.contrib.tpu.python.tpu import tpu_estimator
25 from tensorflow.contrib.tpu.python.tpu import tpu_optimizer
39 """An estimator for Generative Adversarial Networks (GANs) on TPU.
42 but works on TPU.
142 This is ignored for jobs that run on TPU, such as the train job if
163 use_tpu: Same as `TPUEstimator`: A bool indicating whether TPU support is
164 enabled. Currently, TPU training and evaluation respect this bit, bu
    [all...]
  /external/tensorflow/tensorflow/python/tpu/profiler/
tpu_profiler_analysis_pb2_grpc.py 25 from tensorflow.contrib.tpu.profiler import tpu_profiler_analysis_pb2 as third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2
31 TPUProfileAnalysis service provide entry point for profiling TPU and for
74 TPUProfileAnalysis service provide entry point for profiling TPU and for

Completed in 228 milliseconds

12 3 4