| .. |
|
graph_rewrite
|
Fix UpdatedMirroredIndices to take in the additional per_replica inputs and not the new total per replica inputs.
|
2020-10-16 17:18:44 -07:00 |
|
kernels
|
Fix shape inference call in compile op
|
2020-10-14 23:09:57 -07:00 |
|
ops
|
Add target environment constraints to a subset of TensorFlow packages and targets.
|
2020-09-21 10:57:01 -07:00 |
|
BUILD
|
[XLA:PJRT] Add TPU client and associated libtpu.so initializer.
|
2020-10-06 15:15:22 -07:00 |
|
libtftpu.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_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 a new parameter to turn on and off library initialization in TPU APIs
|
2020-09-28 13:56:40 -07:00 |
|
tpu_api_dlsym_initializer.h
|
Group together execution-related C API methods.
|
2020-07-16 16:12:13 -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
|
Separate out the TPU executor API from the rest to minimize build dependencies.
|
2020-08-17 17:58:15 -07:00 |
|
tpu_api.h
|
Separate out the TPU executor API from the rest to minimize build dependencies.
|
2020-08-17 17:58:15 -07:00 |
|
tpu_compilation_device.cc
|
Add some missing dependencies so that the TPU version of TensorFlow builds
|
2020-07-27 17:43:48 -07:00 |
|
tpu_compile_interface.cc
|
TPU rewrite pass refactoring.
|
2020-07-14 14:45:58 -07:00 |
|
tpu_compile_interface.h
|
TPU rewrite pass refactoring.
|
2020-07-14 14:45:58 -07:00 |
|
tpu_config_c_api.h
|
Add Remote Cache support for POD use cases.
|
2020-09-02 23:44:15 -07: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
|
Add a set of dynamic embedding optimizers directly taking an HloModule.
|
2020-09-29 18:51:01 -07:00 |
|
tpu_embedding_optimization_parameters_utils.h
|
Frequency estimator implementation for TPU embedding.
|
2020-09-25 19:43:49 -07: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
|
Adjust TPU build dependencies.
|
2020-07-27 15:37:03 -07:00 |
|
tpu_execute.h
|
Add a TPUExecute function.
|
2020-07-16 23:20:08 -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.cc
|
[XLA:PJRT] Add TPU client and associated libtpu.so initializer.
|
2020-10-06 15:15:22 -07:00 |
|
tpu_executor_init_fns.inc
|
Register HostCount and ChipsPerHost API to executor_fn.
|
2020-10-08 14:49:06 -07: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_library_init_fns.inc
|
Removing the unused HostComputeMetadata field.
|
2020-10-13 22:29:14 -07: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_node_device.cc
|
Fix missing status check.
|
2020-07-21 18:07:46 -07:00 |
|
tpu_node_device.h
|
Introduce tpu_node_device to represent the individual TPU cores
|
2020-07-09 18:15:58 -07:00 |
|
tpu_on_demand_compiler.cc
|
[mlir] Add FusionOp to XLA HLO -> LMHLO
|
2020-10-13 14:20:52 -07:00 |
|
tpu_system_device.cc
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_system_device.h
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07: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 |
