| .. |
|
client
|
Remove deprecated tfrt_enabled test target flag.
|
2020-10-22 12:55:06 -07:00 |
|
experimental
|
|
|
|
ops
|
Fix gradient computation bug in TPU embedding mid-level API.
|
2020-12-15 20:12:54 -08:00 |
|
profiler
|
Pass TPU worker host:port pairs to trace() from capture_tpu_profile.
|
2020-11-05 16:17:46 -08: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
|
|
|
|
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
|
Merge pull request #45166 from ROCmSoftwarePlatform:google_upstream_rocm_add_remove_no_rocm_tag
|
2020-12-21 00:32:48 -08:00 |
|
datasets_test.py
|
Move TPU python files to TF core.
|
2019-02-20 13:07:14 -08:00 |
|
datasets.py
|
|
|
|
device_assignment.py
|
Add an option to assign devices without forming a ring
|
2020-09-11 12:03:04 -07:00 |
|
error_handling.py
|
|
|
|
feature_column_test.py
|
Replace instances of "blacklist" with "denylist" where possible. See Google Developer guidelines at https://developers.google.com/style/word-list#blacklist for more information.
|
2020-07-20 16:05:25 -07:00 |
|
feature_column_v2_test.py
|
Change not exposed in public API
|
2020-07-31 13:16:45 -07:00 |
|
feature_column_v2.py
|
Change not exposed in public API
|
2020-07-31 13:16:45 -07:00 |
|
feature_column.py
|
Change not exposed in public API
|
2020-07-31 13:16:45 -07:00 |
|
functional.py
|
|
|
|
preempted_hook.py
|
|
|
|
session_support.py
|
Increase the ping timeout from 5 seconds to a minute. This gives a bit more breathing room in case the workers somehow become unresponsive .
|
2020-11-19 17:16:05 -08:00 |
|
tensor_tracer_flags.py
|
Tensor Tracer: enable summary mode for models using low level APIs.
|
2020-10-28 12:27:54 -07:00 |
|
tensor_tracer_report.py
|
Removing some unused options from tensor tracer.
|
2020-09-24 16:13:26 -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: enable summary mode for models using low level APIs.
|
2020-10-28 12:27:54 -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
|
Fix gradient computation bug in TPU embedding mid-level API.
|
2020-12-15 20:12:54 -08:00 |
|
tpu_embedding_v2_correctness_test.py
|
Remove shutdown from tpu embedding tests.
|
2020-12-08 09:51:45 -08:00 |
|
tpu_embedding_v2_cpu_test.py
|
Allow delayed creation of variables in TPU Embedding mid level API.
|
2020-07-29 16:25:14 -07:00 |
|
tpu_embedding_v2_test.py
|
Remove shutdown from tpu embedding tests.
|
2020-12-08 09:51:45 -08:00 |
|
tpu_embedding_v2_utils_test.py
|
Split TPUEmbeddingConfiguration logging across multiple statements.
|
2020-10-22 14:46:54 -07:00 |
|
tpu_embedding_v2_utils.py
|
Split TPUEmbeddingConfiguration logging across multiple statements.
|
2020-10-22 14:46:54 -07:00 |
|
tpu_embedding_v2.py
|
fix typos in python directory
|
2020-10-29 16:21:24 +03:00 |
|
tpu_embedding.py
|
Add support for FrequencyEstimator optimizer to the mid level tpu embedding API.
|
2020-12-01 17:02:39 -08:00 |
|
tpu_estimator.py
|
|
|
|
tpu_feed.py
|
Add types to tpu.py.
|
2020-11-03 14:01:22 -08:00 |
|
tpu_function.py
|
Adds a reminder to set export_saved_model_api_version=ExportSavedModelApiVersion.V2 when using TPUEstimator's inference_on_tpu.
|
2020-12-02 16:05:59 -08:00 |
|
tpu_infeed_test.py
|
Move TPU python files to TF core.
|
2019-02-20 13:07:14 -08:00 |
|
tpu_name_util.py
|
Add types to tpu.py.
|
2020-11-03 14:01:22 -08:00 |
|
tpu_optimizer.py
|
|
|
|
tpu_outside_compilation_test.py
|
Add outside compilation test with TPUPartitionedCall.
|
2020-10-30 08:50:52 -07:00 |
|
tpu_sharding_test.py
|
Add number_of_partitions to the InfeedQueue API to allow infeed to be processed with pure data parallelism, but the partition/resharding happens inside the model function.
|
2020-07-30 13:10:17 -07:00 |
|
tpu_sharding.py
|
Add number_of_partitions to the InfeedQueue API to allow infeed to be processed with pure data parallelism, but the partition/resharding happens inside the model function.
|
2020-07-30 13:10:17 -07:00 |
|
tpu_strategy_util.py
|
Allow tpu.experimental.initialize_tpu_system to be called inside tf.function.
|
2020-10-07 22:19:14 -07:00 |
|
tpu_system_metadata.py
|
|
|
|
tpu_test_wrapper_test.py
|
|
|
|
tpu_test_wrapper.bzl
|
Create no-src target for expanding visibility of TF deps.
|
2020-07-20 20:12:23 -07:00 |
|
tpu_test_wrapper.py
|
|
|
|
tpu_test.py
|
Do not cache external values in TPUReplicateContext.AddValue if there is no outer context since that is no-op. Doing so leads to name collisions in the present of nested graphs since op names are not necessarily unique across graphs.
|
2020-09-09 09:20:07 -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
|
Correctly handle the case if static maximum dimension size = 0.
|
2020-12-04 18:17:13 -08:00 |
|
training_loop.py
|
De-dup tpu/xla.py compiler/xla/xla.py
|
2019-04-02 14:21:06 -07:00 |
|
util.py
|
|
|
