experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/python/tpu
History
Bruce Fontaine b878ae3918 Allow tf.distribute.TPUStrategy to be used with TPUEmbedding API and ensure that LossScaleOptimizer properly rejects it. 
PiperOrigin-RevId: 318186211
Change-Id: Id3b9cb8288e5d28ddbaec97d5b35627ab35bc08d
2020-06-25 15:32:43 -07:00
..
client
Support fetching configured runtime version without using Cloud TPU API.
2020-06-17 08:06:10 -07:00
experimental
…
ops
Add EnqueueTPUEmbeddingRaggedTensorBatch for RaggedTensor support.
2020-04-01 13:28:05 -07:00
profiler
switch capture_tpu_profile to new api of profiler_client. because some options is dropped, therefore two flags are deprecated.
2020-05-21 14:21:24 -07:00
__init__.py
…
_tpu_estimator_embedding.py
Delete the TF core copy of TPUEstimator in favor of the Estimator repo's
2019-05-09 14:53:21 -07:00
api.py
Add TPUEmbedding mid level API.
2020-05-19 10:00:34 -07:00
async_checkpoint_test.py
Add test for AsyncCheckpointSaverHook without listeners.
2019-12-18 12:29:43 -08:00
async_checkpoint.py
Fix AsyncCheckpointSaverHook info logging if listeners is None.
2019-12-18 12:01:52 -08:00
bfloat16_test.py
Merge pull request #39597 from Intel-tensorflow:cuixiaom/bf16_namescope_pr
2020-06-03 13:35:44 -07:00
bfloat16.py
[INTEL MKL] Changed the default value of input parameter to None to align with API policy
2020-05-27 11:10:44 -07:00
BUILD
Create TPU outside compilation tests.
2020-06-03 17:20:43 -07:00
datasets_test.py
…
datasets.py
…
device_assignment.py
[TF:TPU] Add 3D mesh support in TPU configuration ops.
2020-03-11 16:11:20 -07:00
error_handling.py
…
feature_column_test.py
…
feature_column_v2_test.py
Enable TensorCore embeddings for training via FeatureColumnV2.
2020-05-19 13:28:12 -07:00
feature_column_v2.py
Fix pylint errors
2020-05-19 14:44:14 -07:00
feature_column.py
Update documentation of tf.tpu.experimental.(shared_)embedding_column to reflect that a user should pass in the same function if they use the same dynamic learning rate for multiple tables.
2020-03-09 16:53:12 -07:00
functional.py
…
preempted_hook.py
Refactor Tpu Cluster Resolver.
2019-11-26 12:08:35 -08:00
session_support.py
TPUEstimator: Fix shutdown behavior after preemption.
2020-03-04 20:12:29 -08:00
tensor_tracer_flags.py
Adding an option to tensor tracer to create a suffix folder based on the fingerprint of the tf.graph.
2020-05-15 18:10:39 -07:00
tensor_tracer_report.py
Adding an option to tensor tracer to create a suffix folder based on the fingerprint of the tf.graph.
2020-05-15 18:10:39 -07:00
tensor_tracer.proto
Adding an option to tensor tracer to create a suffix folder based on the fingerprint of the tf.graph.
2020-05-15 18:10:39 -07:00
tensor_tracer.py
Tensor tracer python3 compatibility.
2020-06-04 21:59:24 -07:00
topology_test.py
[TF:TPU] Add 3D mesh support in TPU configuration ops.
2020-03-11 16:11:20 -07:00
topology.py
Add get_tpu_system_metadata API to TPUClusterResolver. Also export tf.tpu.experimental.TPUSystemMetadata and tf.tpu.experimental.Topology symbols.
2020-03-18 19:48:04 -07:00
tpu_config.py
…
tpu_context.py
…
tpu_embedding_gradient.py
Merge pull request #34985 from kiszk:spelling_tweaks_python
2020-02-12 13:41:49 -08:00
tpu_embedding_v2_cpu_test.py
Add unit tests for TPUEmbedding API.
2020-06-01 13:49:50 -07:00
tpu_embedding_v2_test.py
Fix error check in TPUEmbedding to work when used in outside compilation.
2020-06-17 11:11:11 -07:00
tpu_embedding_v2_utils.py
Allow checkpoint loading before mid level TPU embedding API creation. This should avoid double initialization.
2020-06-04 11:49:48 -07:00
tpu_embedding_v2.py
Allow tf.distribute.TPUStrategy to be used with TPUEmbedding API and ensure that LossScaleOptimizer properly rejects it.
2020-06-25 15:32:43 -07:00
tpu_embedding.py
Split tensorflow.python.tpu module doctests into different target.
2020-05-11 11:57:16 -07:00
tpu_estimator.py
…
tpu_feed.py
Correct typo and better doc.
2020-04-28 14:38:24 -07:00
tpu_function.py
…
tpu_infeed_test.py
…
tpu_optimizer.py
Remove the optimizer v2 instance check, and replace with v1 object check.
2020-03-24 12:17:37 -07:00
tpu_outside_compilation_test.py
Add more outside compilation tests including multiple clusters and more variety of inputs/outputs.
2020-06-18 16:21:10 -07:00
tpu_sharding_test.py
…
tpu_sharding.py
…
tpu_strategy_util.py
Add a TPUClusterResolver.connect API to simplify TPU initialization.
2020-06-20 00:09:59 -07:00
tpu_system_metadata.py
Add get_tpu_system_metadata API to TPUClusterResolver. Also export tf.tpu.experimental.TPUSystemMetadata and tf.tpu.experimental.Topology symbols.
2020-03-18 19:48:04 -07:00
tpu_test_wrapper_test.py
Remove the "/" after path in assertion, as path join used somewhere in windows inserts "\" and breaks assertion.
2020-01-10 13:06:30 -08:00
tpu_test_wrapper.bzl
Pull out tpu_test_wrapper Starlark logic for reuse.
2020-06-02 21:20:40 -07:00
tpu_test_wrapper.py
Fix the ast.Module call for py3.8
2020-02-27 10:53:45 -08:00
tpu_test.py
Allow tf2xla Einsum Op to accept unary inputs.
2019-10-02 01:04:18 -07:00
tpu.bzl
Add an option to enable MLIR bridge for tpu_py_test rule
2020-06-18 13:58:47 -07:00
tpu.py
Support packed variable in DistributedVariable. Add an option to enable packed variable in TPUStrategy.
2020-06-18 20:12:02 -07:00
training_loop.py
De-dup tpu/xla.py compiler/xla/xla.py
2019-04-02 14:21:06 -07:00
util.py
…