• Home
Name Date Size #Lines LOC

..--

client/03-May-2024-1,5001,220

experimental/03-May-2024-4220

ops/03-May-2024-555424

profiler/03-May-2024-599411

BUILDD03-May-202420.2 KiB725680

__init__.pyD03-May-2024845 214

_tpu_estimator_embedding.pyD03-May-20241 KiB245

api.pyD03-May-20241.3 KiB3414

async_checkpoint.pyD03-May-20248.5 KiB219166

async_checkpoint_test.pyD03-May-20247.2 KiB197152

bfloat16.pyD03-May-20243.2 KiB9355

bfloat16_test.pyD03-May-20243.1 KiB8658

datasets.pyD03-May-20247.7 KiB200149

datasets_test.pyD03-May-20247.4 KiB215153

device_assignment.pyD03-May-202420.9 KiB557418

error_handling.pyD03-May-20241,015 245

feature_column.pyD03-May-202429.5 KiB723574

feature_column_test.pyD03-May-202413.1 KiB317243

feature_column_v2.pyD03-May-202446.5 KiB1,093868

feature_column_v2_test.pyD03-May-202428.2 KiB730579

functional.pyD03-May-2024957 246

preempted_hook.pyD03-May-20243.2 KiB9460

session_support.pyD03-May-202414.8 KiB453333

tensor_tracer.protoD03-May-20242.8 KiB7963

tensor_tracer.pyD03-May-202485.3 KiB2,0511,616

tensor_tracer_flags.pyD03-May-202416.6 KiB470379

tensor_tracer_report.pyD03-May-202417 KiB431345

topology.pyD03-May-20249.5 KiB236178

topology_test.pyD03-May-20241.6 KiB4621

tpu.bzlD03-May-20241.7 KiB5448

tpu.pyD03-May-202493.4 KiB2,2711,701

tpu_config.pyD03-May-20241,011 245

tpu_context.pyD03-May-20241,012 245

tpu_embedding.pyD03-May-2024119.2 KiB2,8922,438

tpu_embedding_gradient.pyD03-May-20247.3 KiB180132

tpu_embedding_v2.pyD03-May-202474.3 KiB1,7551,272

tpu_embedding_v2_correctness_test.pyD03-May-202426.5 KiB652478

tpu_embedding_v2_cpu_test.pyD03-May-202414.2 KiB359271

tpu_embedding_v2_test.pyD03-May-202451.2 KiB1,323957

tpu_embedding_v2_utils.pyD03-May-202430.5 KiB752621

tpu_embedding_v2_utils_test.pyD03-May-20244.5 KiB12383

tpu_estimator.pyD03-May-20241.8 KiB3414

tpu_feed.pyD03-May-202438.4 KiB956785

tpu_function.pyD03-May-20242.5 KiB7437

tpu_infeed_test.pyD03-May-20245.6 KiB131106

tpu_name_util.pyD03-May-20241.2 KiB3616

tpu_optimizer.pyD03-May-20248.6 KiB225170

tpu_outside_compilation_test.pyD03-May-202422.2 KiB721535

tpu_sharding.pyD03-May-202411.2 KiB308240

tpu_sharding_test.pyD03-May-20245.2 KiB150119

tpu_strategy_util.pyD03-May-20249.2 KiB220138

tpu_system_metadata.pyD03-May-20248.7 KiB232176

tpu_test.pyD03-May-20247.3 KiB209150

tpu_test_wrapper.bzlD03-May-20243.3 KiB10888

tpu_test_wrapper.pyD03-May-20247 KiB200131

tpu_test_wrapper_test.pyD03-May-20246.5 KiB214143

training_loop.pyD03-May-20249.2 KiB237174

util.pyD03-May-20241,005 245