..
|
graph_rewrite
|
kernels
|
ops
|
BUILD
|
build_defs.bzl
|
tpu_api_dlsym_initializer.cc
|
tpu_api_dlsym_initializer_windows.cc
|
tpu_compile.cc
|
tpu_compile.h
|
tpu_compile_interface.cc
|
tpu_compile_interface.h
|
tpu_configuration.cc
|
tpu_configuration.h
|
tpu_defs.cc
|
tpu_defs.h
|
tpu_embedding_configuration_proto_rewrite.cc
|
tpu_embedding_configuration_proto_rewrite.h
|
tpu_embedding_configuration_proto_rewrite_test.cc
|
tpu_embedding_configuration_utils.cc
|
tpu_embedding_configuration_utils.h
|
tpu_embedding_errors.cc
|
tpu_embedding_errors.h
|
tpu_embedding_errors_test.cc
|
tpu_embedding_optimization_parameters_utils.cc
|
tpu_embedding_optimization_parameters_utils.h
|
tpu_embedding_output_layout_utils.cc
|
tpu_embedding_output_layout_utils.h
|
tpu_embedding_spmd_sharding_utils.cc
|
tpu_embedding_spmd_sharding_utils.h
|
tpu_execute.cc
|
tpu_execute.h
|
tpu_fingerprint_utils.cc
|
tpu_fingerprint_utils.h
|
tpu_global_init.cc
|
tpu_global_init.h
|
tpu_init_mode.cc
|
tpu_init_mode.h
|
tpu_model_server_initializer.cc
|
tpu_model_server_initializer.h
|
tpu_node_device_util.cc
|
tpu_node_device_util.h
|
virtual_device.cc
|
virtual_device.h
|