| .. |
|
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
|
|
|
|
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
|
Move TPU python files to TF core.
|
2019-02-20 13:07:14 -08:00 |
|
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
|
|
|
|
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
|
|
|
|
util.py
|
|
|
