| .. |
|
client
|
Enable more TFRT test targets that are newly passing
|
2020-09-23 11:44:50 -07:00 |
|
experimental
|
|
|
|
ops
|
|
|
|
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
|
|
|
|
async_checkpoint.py
|
|
|
|
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
|
Removed the unnecessary deps of Keras in control_flow_v2.
|
2020-10-08 13:40:08 -07:00 |
|
datasets_test.py
|
|
|
|
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
|
|
|
|
tensor_tracer_flags.py
|
TensorTracer: Removing the necessity of dumping traces to get meaningful reads.
|
2020-10-02 17:54:40 -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 file updates.
|
2020-10-05 10:52:07 -07:00 |
|
topology_test.py
|
|
|
|
topology.py
|
|
|
|
tpu_config.py
|
|
|
|
tpu_context.py
|
|
|
|
tpu_embedding_gradient.py
|
|
|
|
tpu_embedding_v2_correctness_test.py
|
Rename "experimental_distribute_datasets_from_function" to "distribute_datasets_from_function".
|
2020-09-23 18:15:32 -07: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
|
Rename "experimental_distribute_datasets_from_function" to "distribute_datasets_from_function".
|
2020-09-23 18:15:32 -07:00 |
|
tpu_embedding_v2_utils_test.py
|
Add sensible __repr__ implementation for TableConfig, FeatureConfig.
|
2020-10-02 11:13:45 -07:00 |
|
tpu_embedding_v2_utils.py
|
Add sensible __repr__ implementation for TableConfig, FeatureConfig.
|
2020-10-02 11:13:45 -07:00 |
|
tpu_embedding_v2.py
|
Make ShardedVariable a composite tensor.
|
2020-10-15 14:32:44 -07:00 |
|
tpu_embedding.py
|
Update tpu_embedding.py
|
2021-03-03 11:36:24 -08:00 |
|
tpu_estimator.py
|
|
|
|
tpu_feed.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_function.py
|
|
|
|
tpu_infeed_test.py
|
|
|
|
tpu_optimizer.py
|
|
|
|
tpu_outside_compilation_test.py
|
Ensure that if an op is outside compiled, its gradient gets outside compiled as well in TF2 (similar to TF1) when colocate_gradients_with_ops=True.
|
2020-10-15 15:47:50 -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
|
Ensure that if an op is outside compiled, its gradient gets outside compiled as well in TF2 (similar to TF1) when colocate_gradients_with_ops=True.
|
2020-10-15 15:47:50 -07:00 |
|
training_loop.py
|
|
|
|
util.py
|
|
|
