| .. |
|
xla
|
[XLA] Break dependencies of XLA on TF.
|
2021-01-19 14:07:56 -08:00 |
|
BUILD
|
Add a few utilities for fingerprint.
|
2021-01-27 14:47:26 -08:00 |
|
compiled_subgraph.h
|
[Resubmit] Only trigger resharding when necessary.
|
2020-09-16 16:23:31 -07:00 |
|
cross_replica_ops.cc
|
Add cross replica op definitions.
|
2020-07-20 12:23:25 -07:00 |
|
host_compute_ops.cc
|
Update tf._XlaRecvAtHost and tf._XlaSendFromHost to have similar ops that operate on a runtime device ordinal instead of a compile time/constant attribute (tf._XlaRecvAtHostV2 and tf._XlaSendFromHostV2).
|
2020-11-05 16:54:02 -08:00 |
|
image_resize_ops.cc
|
Re-introduce some infeed and outfeed ops after a revert caused by failing tests.
|
2020-08-10 11:10:46 -07:00 |
|
infeed_ops.cc
|
Re-introduce some infeed and outfeed ops after a revert caused by failing tests.
|
2020-08-10 11:10:46 -07:00 |
|
infeed_ops.h
|
Re-introduce some infeed and outfeed ops after a revert caused by failing tests.
|
2020-08-10 11:10:46 -07:00 |
|
outfeed_ops.cc
|
[XLA] Drop useless shape argument from TransferManager::TransferLiteralFromOutfeed.
|
2021-01-25 17:29:36 -08:00 |
|
outfeed_ops.h
|
Add OutfeedDequeue ops that allow for a dynamic device ordinal to be used.
|
2020-09-23 16:00:47 -07:00 |
|
replication_ops.cc
|
Re-introduce some infeed and outfeed ops after a revert caused by failing tests.
|
2020-08-10 11:10:46 -07:00 |
|
topk_ops.cc
|
Add TPU kernel definitions.
|
2020-07-23 17:12:37 -07:00 |
|
tpu_compilation_cache_common.proto
|
core/tpu/kernels/BUILD file proto target refactoring
|
2020-08-24 15:36:26 -07:00 |
|
tpu_compilation_cache_entry_unloader.h
|
Prep change for publishing TPU Ops.
|
2020-06-18 15:05:28 -07:00 |
|
tpu_compilation_cache_entry.h
|
Refactor TpuCompilationCacheEntry interface to return TpuProgramGroupInterface and core_index and makes CacheEntry less transparent and move application specific logics outside of cache.
|
2020-08-05 16:10:21 -07:00 |
|
tpu_compilation_cache_external.cc
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_compilation_cache_external.h
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_compilation_cache_factory.cc
|
Introduce TpuCompilationCache create function registration
|
2020-07-15 11:05:25 -07:00 |
|
tpu_compilation_cache_factory.h
|
Introduce TpuCompilationCache create function registration
|
2020-07-15 11:05:25 -07:00 |
|
tpu_compilation_cache_grpc.cc
|
Internal dependency cleanup.
|
2021-01-11 17:21:12 -08:00 |
|
tpu_compilation_cache_grpc.h
|
Internal dependency cleanup.
|
2021-01-11 17:21:12 -08:00 |
|
tpu_compilation_cache_interface.cc
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_compilation_cache_interface.h
|
[Resubmit] Only trigger resharding when necessary.
|
2020-09-16 16:23:31 -07:00 |
|
tpu_compilation_cache_key.h
|
TPU library internal change.
|
2020-06-12 01:49:35 -07:00 |
|
tpu_compilation_cache_local_lookup.cc
|
Refactor TpuCompilationCacheEntry interface to return TpuProgramGroupInterface and core_index and makes CacheEntry less transparent and move application specific logics outside of cache.
|
2020-08-05 16:10:21 -07:00 |
|
tpu_compilation_cache_local_lookup.h
|
core/tpu/kernels/BUILD file proto target refactoring
|
2020-08-24 15:36:26 -07:00 |
|
tpu_compilation_cache_lookup.h
|
core/tpu/kernels/BUILD file proto target refactoring
|
2020-08-24 15:36:26 -07:00 |
|
tpu_compilation_cache_rpc_lookup.cc
|
Internal dependency cleanup.
|
2021-01-11 17:21:12 -08:00 |
|
tpu_compilation_cache_rpc_lookup.h
|
core/tpu/kernels/BUILD file proto target refactoring
|
2020-08-24 15:36:26 -07:00 |
|
tpu_compilation_cache_rpc_support.cc
|
Removing the unused HostComputeMetadata field.
|
2020-10-13 22:29:14 -07:00 |
|
tpu_compilation_cache_rpc_support.h
|
Internal dependency cleanup.
|
2021-01-11 17:21:12 -08:00 |
|
tpu_compilation_cache_service.cc
|
+ Fix assigning default value to deleted std::atomic<bool> for OSS.
|
2020-09-03 12:06:10 -07:00 |
|
tpu_compilation_cache_service.h
|
+ Fix assigning default value to deleted std::atomic<bool> for OSS.
|
2020-09-03 12:06:10 -07:00 |
|
tpu_compilation_cache.proto
|
core/tpu/kernels/BUILD file proto target refactoring
|
2020-08-24 15:36:26 -07:00 |
|
tpu_compilation_metrics.cc
|
Internal change
|
2020-10-02 20:33:42 -07:00 |
|
tpu_compilation_metrics.h
|
Rename TpuCompilationCacheMetrics into TpuCompilationMetrics and add IncrementCompilationCount.
|
2020-07-07 17:28:17 -07:00 |
|
tpu_compile_op_common.cc
|
Add error payload in status.
|
2021-01-07 00:09:29 -08:00 |
|
tpu_compile_op_common.h
|
Cache a fingerprint of the MLIR module on TPUCompileMlir kernel
|
2020-10-28 23:05:25 -07:00 |
|
tpu_compile_op_impl.cc
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_compile_op_impl.h
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_compile_op_options.cc
|
|
|
|
tpu_compile_op_options.h
|
|
|
|
tpu_compile_op_support.cc
|
Deprecate the experimental C_API CompileAheadOfTime().
|
2020-09-14 17:13:41 -07:00 |
|
tpu_compile_op_support.h
|
Deprecate the experimental C_API CompileAheadOfTime().
|
2020-09-14 17:13:41 -07:00 |
|
tpu_compile_op.cc
|
Add TPU compilation not succeeding warning message
|
2020-11-17 22:02:47 -08:00 |
|
tpu_compile_op.h
|
Add TpuCompileOp implementation.
|
2020-07-20 16:09:55 -07:00 |
|
tpu_compile.proto
|
Deprecate the experimental C_API CompileAheadOfTime().
|
2020-09-14 17:13:41 -07:00 |
|
tpu_configuration_ops.cc
|
[tpu1vm] Make Compilation Cache Server Lookup APIs Forward Compatible
|
2020-11-19 17:23:01 -08:00 |
|
tpu_configuration_ops.h
|
Add Remote Cache support for POD use cases.
|
2020-09-02 23:44:15 -07:00 |
|
tpu_executable_info.proto
|
|
|
|
tpu_execute_op.cc
|
Ask users to set drop_remainder=False in the dataset when encountering runtime mismatch errors on TPUs.
|
2021-01-29 00:27:03 -08:00 |
|
tpu_execute_op.h
|
Add a TPU execution op.
|
2020-07-17 17:52:03 -07:00 |
|
tpu_handle_to_key_op.cc
|
Re-introduce some infeed and outfeed ops after a revert caused by failing tests.
|
2020-08-10 11:10:46 -07:00 |
|
tpu_mesh_state_interface.h
|
[TPU 1VM] Consolidate all TPU ops related APIs into a single file
|
2020-10-23 18:09:32 -07:00 |
|
tpu_op_consts.cc
|
Add a few utilities for fingerprint.
|
2021-01-27 14:47:26 -08:00 |
|
tpu_op_consts.h
|
Add a few utilities for fingerprint.
|
2021-01-27 14:47:26 -08:00 |
|
tpu_op_util.cc
|
Reduce verbosity in a util function.
|
2021-02-03 10:24:08 -08:00 |
|
tpu_op_util.h
|
Add a few utilities for fingerprint.
|
2021-01-27 14:47:26 -08:00 |
|
tpu_pod_state.cc
|
[tpu1vm] Make Compilation Cache Server Lookup APIs Forward Compatible
|
2020-11-19 17:23:01 -08:00 |
|
tpu_pod_state.h
|
Fix server address assignment when constructing TpuPodState.
|
2020-09-03 18:36:49 -07:00 |
|
tpu_program_group_interface.h
|
Implement Serialization/Deserialization for remote compilation cache.
|
2020-09-01 12:51:30 -07:00 |
|
tpu_program_group.cc
|
Add a new CAPI to take xrt_computation and perform the compilation.
|
2020-11-16 16:21:48 -08:00 |
|
tpu_program_group.h
|
Add a new CAPI to take xrt_computation and perform the compilation.
|
2020-11-16 16:21:48 -08:00 |
|
tpu_util.cc
|
Add Remote Cache support for POD use cases.
|
2020-09-02 23:44:15 -07:00 |
|
tpu_util.h
|
Add Remote Cache support for POD use cases.
|
2020-09-02 23:44:15 -07:00 |
|
trace_util.h
|
|
|
|
transfer_ops.cc
|
Return an error instead of crashing
|
2021-01-10 16:31:04 -08:00 |
|
transfer_ops.h
|
Add OutfeedDequeue ops that allow for a dynamic device ordinal to be used.
|
2020-09-23 16:00:47 -07:00 |
