Home
last modified time | relevance | path

Searched refs:tpu (Results 1 – 25 of 136) sorted by relevance

123456

/external/tensorflow/tensorflow/contrib/tpu/
D__init__.py68 from tensorflow.contrib.tpu.python import profiler
69 from tensorflow.contrib.tpu.python.ops.tpu_ops import *
70 from tensorflow.contrib.tpu.python.tpu.async_checkpoint import *
71 from tensorflow.contrib.tpu.python.tpu.bfloat16 import *
72 from tensorflow.contrib.tpu.python.tpu.device_assignment import *
73 from tensorflow.contrib.tpu.python.tpu.keras_support import tpu_model as keras_to_tpu_model
74 from tensorflow.contrib.tpu.python.tpu.keras_support import TPUDistributionStrategy
75 from tensorflow.contrib.tpu.python.tpu.topology import *
76 from tensorflow.contrib.tpu.python.tpu.tpu import *
77 from tensorflow.contrib.tpu.python.tpu.tpu_config import *
[all …]
DBUILD16 "//cloud/vmm/testing/tests/tpu:__subpackages__",
32 "//tensorflow/python/tpu:tpu_py",
38 srcs = ["python/tpu/async_checkpoint.py"],
41 "//tensorflow/python/tpu:async_checkpoint",
48 "python/tpu/_tpu_estimator_embedding.py",
49 "python/tpu/error_handling.py",
50 "python/tpu/tpu_config.py",
51 "python/tpu/tpu_context.py",
52 "python/tpu/tpu_estimator.py",
53 "python/tpu/util.py",
[all …]
DREADME.md14 from tensorflow.contrib.tpu.python.tpu import tpu_config
15 from tensorflow.contrib.tpu.python.tpu import tpu_estimator
16 from tensorflow.contrib.tpu.python.tpu import tpu_optimizer
Dtpu_estimator.md6 [Cloud TPU documentation](https://cloud.google.com/tpu/docs).
172 [github.com/tensorflow/tpu](https://github.com/tensorflow/tpu).
177 - Switch from `tf.estimator.RunConfig` to `tf.contrib.tpu.RunConfig`.
178 - Set the `TPUConfig` (part of the `tf.contrib.tpu.RunConfig`) to specify the
185 - In `model_fn`, use `tf.contrib.tpu.CrossShardOptimizer` to wrap your
191 - Switch from `tf.estimator.Estimator` to `tf.contrib.tpu.TPUEstimator`.
226 …GCP Cloud TPUs refer to the [Cloud Documentation](https://cloud.google.com/tpu/docs/tutorials/mnis…
227 Refer to this link for all [Cloud TPU documentation](https://cloud.google.com/tpu/docs).
231 …tructions as specified in the [Cloud TPU Tools](https://cloud.google.com/tpu/docs/cloud-tpu-tools).
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dtpu_estimator.py22 from tensorflow.python.tpu.tpu_estimator import *
24 from tensorflow.python.tpu.tpu_estimator import _clone_export_output_with_tensors
25 from tensorflow.python.tpu.tpu_estimator import _create_global_step
26 from tensorflow.python.tpu.tpu_estimator import _export_output_to_tensors
27 from tensorflow.python.tpu.tpu_estimator import _get_scaffold
28 from tensorflow.python.tpu.tpu_estimator import _Inputs
29 from tensorflow.python.tpu.tpu_estimator import _ITERATIONS_PER_LOOP_VAR
30 from tensorflow.python.tpu.tpu_estimator import _TPU_ENQUEUE_OPS
31 from tensorflow.python.tpu.tpu_estimator import _TPU_ESTIMATOR
32 from tensorflow.python.tpu.tpu_estimator import _TPU_TRAIN_OP
Dfeature_column.py22 from tensorflow.python.tpu.feature_column import *
24 from tensorflow.python.tpu.feature_column import _is_running_on_cpu
25 from tensorflow.python.tpu.feature_column import _record_variable_scope_and_name
26 from tensorflow.python.tpu.feature_column import _TPU_FC_TO_SCOPE
27 from tensorflow.python.tpu.feature_column import _TPUBaseEmbeddingColumn
28 from tensorflow.python.tpu.feature_column import _TPUEmbeddingColumn
29 from tensorflow.python.tpu.feature_column import _TPUSharedEmbeddingColumn
Dtpu.py22 from tensorflow.python.tpu.tpu import *
24 from tensorflow.python.tpu.tpu import _TPU_REPLICATE_ATTR
Dtpu_system_metadata.py22 from tensorflow.python.tpu.tpu_system_metadata import *
24 from tensorflow.python.tpu.tpu_system_metadata import _query_tpu_system_metadata
Dtpu_feed.py22 from tensorflow.python.tpu.tpu_feed import *
24 from tensorflow.python.tpu.tpu_feed import _PartitionedInfeedQueue
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dtpu_cluster_resolver.py181 tpu=None, argument
232 if isinstance(tpu, list):
233 if not tpu:
235 if len(tpu) != 1:
238 tpu = tpu[0]
242 if tpu is None:
244 tpu = self._gkeEndpoints()
246 tpu = self._envVarFallback()
248 if tpu is None:
251 self._tpu = compat.as_bytes(tpu) # self._tpu is always bytes
[all …]
Dtpu_cluster_resolver_test.py139 TPUClusterResolver(tpu='')
168 tpu=['test-tpu-1'],
202 tpu=['test-tpu-1'],
229 tpu='test-tpu-1',
249 tpu=['test-tpu-1'],
277 tpu='test-tpu-1',
321 tpu='test-tpu-1',
372 tpu='test-tpu-1',
397 tpu=[],
403 def verifyShouldResolve(self, tpu, should_resolve): argument
[all …]
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_ops.cc77 tpu::OptimizationAlgorithm alg, bool is_debug_op, in RegisterPerTableLoadOpsForAlgorithmBody()
79 tpu::GradientAccumulationSupport grad_accum_support; in RegisterPerTableLoadOpsForAlgorithmBody()
82 std::vector<tpu::StateVariableSpecification> state_variable_specs; in RegisterPerTableLoadOpsForAlgorithmBody()
85 grad_accum_support == tpu::GradientAccumulationSupport::kSupported && in RegisterPerTableLoadOpsForAlgorithmBody()
158 [&](const tpu::StateVariableSpecification& sv) { in RegisterPerTableLoadOpsForAlgorithmBody()
193 tpu::OptimizationAlgorithm alg, bool is_debug_op, in RegisterPerTableRetrieveOpsForAlgorithmBody()
195 tpu::GradientAccumulationSupport grad_accum_support; in RegisterPerTableRetrieveOpsForAlgorithmBody()
198 std::vector<tpu::StateVariableSpecification> state_variable_specs; in RegisterPerTableRetrieveOpsForAlgorithmBody()
201 grad_accum_support == tpu::GradientAccumulationSupport::kSupported && in RegisterPerTableRetrieveOpsForAlgorithmBody()
207 "RetrieveTPUEmbedding", tpu::GetOptimizationAlgorithmName(alg), in RegisterPerTableRetrieveOpsForAlgorithmBody()
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dtpu_test.py35 from tensorflow.python.tpu import tpu
36 from tensorflow.python.tpu import tpu_feed
37 from tensorflow.python.tpu import training_loop
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)
130 tpu._TPU_REPLICATE_ATTR)
137 tpu._TPU_REPLICATE_ATTR)
[all …]
DBUILD16 "//cloud/vmm/testing/tests/tpu:__subpackages__",
109 name = "tpu",
131 "tpu.py",
149 "//tensorflow/core/protobuf/tpu:compilation_result_proto_py",
150 "//tensorflow/core/protobuf/tpu:dynamic_padding_proto_py",
151 "//tensorflow/core/protobuf/tpu:optimization_parameters_proto_py",
152 "//tensorflow/core/protobuf/tpu:topology_proto_py",
153 "//tensorflow/core/protobuf/tpu:tpu_embedding_configuration_proto_py",
154 "//tensorflow/core/protobuf/tpu:tpu_embedding_output_layout_proto_py",
167 "//tensorflow/python/tpu/profiler",
[all …]
Dtpu_strategy_util.py30 from tensorflow.python.tpu import functional as tpu_functional_ops
31 from tensorflow.python.tpu import topology
32 from tensorflow.python.tpu import tpu
81 return tpu.initialize_system()
100 serialized_topology = sess.run(tpu.initialize_system())
Dfeature_column.py27 from tensorflow.python.tpu import tpu
28 from tensorflow.python.tpu import tpu_function
280 if tpu.under_tpu_inference_context():
284 return tpu.outside_compilation(host_computation)
378 if tpu.under_tpu_inference_context():
382 return tpu.outside_compilation(host_computation)
D_tpu_estimator_embedding.py26 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
Dtraining_loop.py25 from tensorflow.python.tpu import tensor_tracer
26 from tensorflow.python.tpu import tpu_function
27 from tensorflow.python.tpu import xla
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dmain_estimator_tpu.py138 optimizer = tf.contrib.tpu.CrossShardOptimizer(optimizer)
156 return tf.contrib.tpu.TPUEstimatorSpec(
163 return tf.contrib.tpu.TPUEstimatorSpec(
173 return tf.contrib.tpu.TPUEstimatorSpec(
194 FLAGS.tpu, zone=FLAGS.tpu_zone, project=FLAGS.gcp_project)
197 config = tf.contrib.tpu.RunConfig(
202 tpu_config=tf.contrib.tpu.TPUConfig(
205 per_host_input_for_training=tf.contrib.tpu.InputPipelineConfig.
219 revnet_classifier = tf.contrib.tpu.TPUEstimator(
/external/tensorflow/tensorflow/core/tpu/
DBUILD13 "//tensorflow/core/protobuf/tpu:optimization_parameters_proto_cc",
27 "//tensorflow/core/protobuf/tpu:tpu_embedding_configuration_proto_cc",
28 "//tensorflow/core/protobuf/tpu:tpu_embedding_output_layout_proto_cc",
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py43 from tensorflow.python.tpu import device_assignment as device_assignment_lib
44 from tensorflow.python.tpu import tpu
45 from tensorflow.python.tpu import tpu_strategy_util
46 from tensorflow.python.tpu import tpu_system_metadata as tpu_system_metadata_lib
47 from tensorflow.python.tpu import training_loop
48 from tensorflow.python.tpu.ops import tpu_ops
181 replicate_outputs = tpu.replicate(replicated_fn, replicate_inputs)
330 replicate_outputs = tpu.replicate(run_fn, replicate_inputs)
643 return (tpu.core(0),)
/external/tensorflow/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/
Dmain.py93 if not FLAGS.service_addr and not FLAGS.tpu:
98 if FLAGS.tpu:
105 [FLAGS.tpu], zone=FLAGS.tpu_zone, project=FLAGS.gcp_project))
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dtpu_gan_estimator_test.py31 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
/external/tensorflow/tensorflow/contrib/cmake/
Dpython_protos.txt4 tensorflow/core/protobuf/tpu
16 tensorflow/contrib/tpu/profiler
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ConfigureDistributedTPU.pbtxt7 A serialized tensorflow.tpu.TopologyProto that describes the TPU
20 Serialized tensorflow.tpu.TPUEmbeddingConfiguration that

123456