100644 blob 3bba5e5ef9b44ffa12571d95a24394c4e6aab2d8	BUILD
100644 blob 2652b57cb6e989ef79e659ae876d610785e1420b	__init__.py
100644 blob 196c8779658fa7cf19a45cabcee8c76085f2a077	api.py
100644 blob 768e46e79e0941d8c338c742fe2dcad9b0a3b48b	async_checkpoint.py
100644 blob 070eff0e20c60e778e49377669fd250532f43a88	async_checkpoint_test.py
100644 blob 284695158f8fc35fdbd14abdb1a1ac96d3ce43bf	bfloat16.py
100644 blob 480c4b286d450bd86624f1df6c5541022615e368	bfloat16_test.py
040000 tree 4de7a2243a59143620178e3a19e37041b1cd9b04	client
100644 blob 3481acc0eb895c154652e246dcb99b86e7c15d0d	datasets.py
100644 blob 6d39731e53a29fb6b45514fa9864704bcf0a9da1	datasets_test.py
100644 blob f73064f7b7a9daf8dccd16ffa7b8b4d1a6869ec6	device_assignment.py
100644 blob 1e6660af511bc1c7a4bd0fe405c95e5c0a6a77bf	error_handling.py
040000 tree 73795eb0eef9859df7a0c6ae1d854da49ddb9a78	experimental
100644 blob 2dffcebf24e88e469909a97751fb7ecac4c7bf04	feature_column.py
100644 blob 1953401a5bcbfad8a4756193c502eb390adb1da6	feature_column_test.py
100644 blob e1b8f8012747305cf99458246d9a80f9a03172ba	feature_column_v2.py
100644 blob f278295c27683a4b815c38ebba2308cc2feb3f21	feature_column_v2_test.py
100644 blob 6afdff26fca3ecc41951a4ca3f86a961b9de42ff	functional.py
040000 tree 2fadfac99f4ec7d08d49885460ee3cc713f260dc	ops
100644 blob c9bedb9343e76295ddf4c4db064f65768aebdd1f	preempted_hook.py
040000 tree 6e0c7ea776c0e111b30ffa6d1b193df5725505fd	profiler
100644 blob 0c2a337a2a68aaa6d91c9e7f0501372dcb3c4b86	session_support.py
100644 blob 664dcb9aaef3583614cf1615ffbfef7facaaf03f	tensor_tracer.proto
100644 blob 8a2e69e46961c5d290cb9b9c66d62216d30a4e78	tensor_tracer.py
100644 blob e9617ce4178774278d6a48d0cd5a8946bfc6ab4e	tensor_tracer_flags.py
100644 blob 2d47a7ddb258be37107865c7f8104df83bee5e26	tensor_tracer_report.py
040000 tree 26fa356a27b196e1a8b25d5d1c0a6b1929618a44	tests
100644 blob 49bdd996c129fa15e008db7d9be4947f78d60fdb	topology.py
100644 blob bc1a3a8126f50b4cc1699661bc99ebd821e35d23	topology_test.py
100644 blob cdc33d35786840c6b8c1842689562041c066a04b	tpu.bzl
100644 blob 2daee6c2a96cbc370da647f8c643f12e6d508275	tpu.py
100644 blob eda3717520f7a878b2f4017968e628c69d402aa7	tpu_config.py
100644 blob d1f3ee55723df3767ae4b0383b3f59f45d3907dc	tpu_context.py
100644 blob 1ef23a0b6242f7715245e51846f78ed73317907a	tpu_embedding.py
100644 blob 1f064359d77631b6de47634eeab7954b477cdf49	tpu_embedding_base.py
100644 blob 9914e084bb1f379ce9fbf5def259c43b14cf0a71	tpu_embedding_for_serving.py
100644 blob bd84f0048b28189a522f14675760f7185c9b8003	tpu_embedding_for_serving_test.py
100644 blob cb272c296341bbae1acb5b960e55b23f4149a109	tpu_embedding_gradient.py
100644 blob 7b19500025bbc13f5daed1f39706bcf8b47827c7	tpu_embedding_v1.py
100644 blob cc653d313673913546d4f3b7f577c9993f2a3a7d	tpu_embedding_v2.py
100644 blob 6641f23e221041b2e38ce3a2983394faf48e6643	tpu_embedding_v2_utils.py
100644 blob 9e18c8c3a0cf8882e4bfd9e36f80ad062fd6a3b3	tpu_embedding_v2_utils_test.py
100644 blob f28db848e56252ab3f70a9ba21ff22ccc16410a9	tpu_estimator.py
100644 blob 23441ccaaad413bc60b3aeaa181907beabe282f1	tpu_feed.py
100644 blob 357746f11444fe603b2b499dfa0c8f2255dbcb85	tpu_function.py
100644 blob 438c876f0fe7e6b16bab0e1e6ca39997d4a8d950	tpu_hardware_feature.py
100644 blob 4f2dc122ee2160b01cee698ddb3b4c0483e65c6a	tpu_infeed_test.py
100644 blob a201fe2d20830180b80fdafbbb2f0a861fecc86a	tpu_name_util.py
100644 blob 0140d34aa1687c77be5a5cd41da0cabc3d71fe3f	tpu_optimizer.py
100644 blob eb0ce826ca69c6ee151158a4ec69cb74f3166e60	tpu_outside_compilation_test.py
100644 blob f88dd2f192efa1d12137ee60f81d176f73b87e79	tpu_replication.py
100644 blob 3a78f45e5292931042b4fc1faae7c36006979bce	tpu_sharding.py
100644 blob 6e14e2ec086ea9771710f080f2a60ba7844ee5ee	tpu_sharding_test.py
100644 blob e122d4816ae6b1182d6e590bc961c5a3aa886849	tpu_strategy_util.py
100644 blob 82f906c6160262c0c8bdea7c13bd49232032f0fa	tpu_system_metadata.py
100644 blob 19cf06721679a30301e664e9bfe0c1f16b7f1934	tpu_test.py
100644 blob 48ccdc3b2d1cfb0deb3288f7ce1a01faa3f9fb85	tpu_test_wrapper.bzl
100644 blob 685a4c5d70a1ac02adc908a928a846097bb20cf6	tpu_test_wrapper.py
100644 blob c4396236fb9636b899153a52b5466631c1726371	tpu_test_wrapper_test.py
100644 blob ea379e92b5740cdcd72f3d34a7f6f09d6e0c9c9a	training_loop.py
100644 blob c5b8964b20a6e290558a322edc04d8428dc56eca	util.py