| .. |
|
graph_rewrite
|
Disable ReshardOp for libtpu
|
2021-03-04 15:41:50 -08:00 |
|
kernels
|
Include call_once.h in tpu_functional_ops.h
|
2021-03-04 21:59:19 -08:00 |
|
ops
|
Open source op definition for TPUReshardVariable op
|
2021-02-18 14:43:04 -08:00 |
|
BUILD
|
Change types to fix TF TPU Build.
|
2021-03-01 18:15:49 -08:00 |
|
libtftpu.h
|
[libtpu] Make copies of data returned by getenv.
|
2021-01-28 14:40:13 -08:00 |
|
tpu_api_dlsym_initializer_windows.cc
|
Rename tpu_load_library to tpu_api_dlsym_initializer.
|
2020-06-22 17:45:58 -07:00 |
|
tpu_api_dlsym_initializer.cc
|
Add some missing headers to tpu_api_dlsym_initializer
|
2021-02-24 11:37:27 -08:00 |
|
tpu_api_dlsym_initializer.h
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_api_dlsym_set_fn.h
|
Add a new parameter to turn on and off library initialization in TPU APIs
|
2020-09-28 13:56:40 -07:00 |
|
tpu_api.cc
|
Make profiler C APIs portable by serializing opaque protobufs into a user provided buffer.
|
2021-02-02 18:52:54 -08:00 |
|
tpu_api.h
|
Make profiler C APIs portable by serializing opaque protobufs into a user provided buffer.
|
2021-02-02 18:52:54 -08:00 |
|
tpu_compile_interface.cc
|
TPU rewrite pass refactoring.
|
2020-07-14 14:45:58 -07:00 |
|
tpu_compile_interface.h
|
Add error payload in status.
|
2021-01-07 00:09:29 -08:00 |
|
tpu_configuration.cc
|
Add a global resource manager for TPU specific operations.
|
2020-05-19 17:38:56 -07:00 |
|
tpu_configuration.h
|
Add a global resource manager for TPU specific operations.
|
2020-05-19 17:38:56 -07:00 |
|
tpu_defs.cc
|
Introduce additional TPU infeed and outfeed ops
|
2020-08-10 01:05:31 -07:00 |
|
tpu_defs.h
|
Introduce additional TPU infeed and outfeed ops
|
2020-08-10 01:05:31 -07:00 |
|
tpu_embedding_optimization_parameters_utils.cc
|
Changed API for getting TPU embedding slot variable information and made a related helper function public.
|
2021-03-03 17:05:33 -08:00 |
|
tpu_embedding_optimization_parameters_utils.h
|
Changed API for getting TPU embedding slot variable information and made a related helper function public.
|
2021-03-03 17:05:33 -08:00 |
|
tpu_embedding_output_layout_utils.cc
|
Deprecated and removed uses of TPUEmbeddingOutputLayout proto and output_layout field in TPUEmbeddingConfiguration.
|
2020-09-30 14:37:01 -07:00 |
|
tpu_embedding_output_layout_utils.h
|
Deprecated and removed uses of TPUEmbeddingOutputLayout proto and output_layout field in TPUEmbeddingConfiguration.
|
2020-09-30 14:37:01 -07:00 |
|
tpu_execute.cc
|
Update Configure TPU, Wait For TPU, and Initialize TPU APIs to backward compatible API style
|
2020-11-13 11:34:22 -08:00 |
|
tpu_execute.h
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_executor_api.cc
|
Don't attempt to register "TPU" platform if the underlying C API isn't initialized.
|
2020-09-21 18:04:07 -07:00 |
|
tpu_executor_api.h
|
Don't attempt to register "TPU" platform if the underlying C API isn't initialized.
|
2020-09-21 18:04:07 -07:00 |
|
tpu_executor_dlsym_initializer_windows.cc
|
Merge pull request #44731 from cloudhan:jax_winbuild
|
2020-11-10 19:44:10 -08:00 |
|
tpu_executor_dlsym_initializer.cc
|
Put definitions that are only used locally within compilation unit behind an anonymous namespace to avoid an ODR.
|
2021-02-19 17:53:57 -08:00 |
|
tpu_executor_init_fns.inc
|
Initialize functions for resharding variables.
|
2021-02-25 23:15:54 -08:00 |
|
tpu_fingerprint_utils.cc
|
Refactor tpu function fingerprint to be a separate library
|
2021-01-27 11:15:06 -08:00 |
|
tpu_fingerprint_utils.h
|
Refactor tpu function fingerprint to be a separate library
|
2021-01-27 11:15:06 -08:00 |
|
tpu_init_mode.cc
|
Introduce some common constants for TPU.
|
2020-05-08 19:53:44 -07:00 |
|
tpu_init_mode.h
|
Introduce some common constants for TPU.
|
2020-05-08 19:53:44 -07:00 |
|
tpu_initializer_helper.cc
|
[libtpu] Fix memory management bug in GetLibTpuInitArguments().
|
2021-03-03 19:47:10 -08:00 |
|
tpu_initializer_helper.h
|
Create a lockfile when loading libtpu.so to prevent attempts at double loading and initialization
|
2021-02-14 23:04:45 -08:00 |
|
tpu_library_init_fns.inc
|
Add partitioned call params struct.
|
2021-03-02 11:58:54 -08:00 |
|
tpu_model_server_initializer.cc
|
Change tpu_model_server_initializer to depend on tpu_api_dlsym_initializer .
|
2021-02-23 15:43:52 -08:00 |
|
tpu_model_server_initializer.h
|
Change tpu_model_server_initializer to depend on tpu_api_dlsym_initializer .
|
2021-02-23 15:43:52 -08:00 |
|
tpu_node_device_util.cc
|
Add a few more utility functions for TPUs
|
2020-05-13 17:07:28 -07:00 |
|
tpu_node_device_util.h
|
Add a few more utility functions for TPUs
|
2020-05-13 17:07:28 -07:00 |
|
tpu_on_demand_compiler.cc
|
Fix bug in tpu_on_demand_compiler.cc
|
2021-01-22 17:30:07 -08:00 |
|
tpu_ops_c_api.h
|
Add partitioned call params struct.
|
2021-03-02 11:58:54 -08:00 |
|
virtual_device.cc
|
Add virtual_device to TPU library
|
2020-06-26 15:51:39 -07:00 |
|
virtual_device.h
|
Add virtual_device to TPU library
|
2020-06-26 15:51:39 -07:00 |
