| .. |
|
BUILD
|
Add HostCount() / ChipsPerHost() API to TPUTopologyExternal.
|
2020-10-07 15:39:48 -07:00 |
|
c_api_conversions.cc
|
[libtpu] Fix static_asserts.
|
2020-10-19 21:34:13 -07:00 |
|
c_api_conversions.h
|
[libtpu] Reimplement XLA_Shape in C API for performance.
|
2020-10-19 14:17:22 -07:00 |
|
c_api_decl.h
|
[libtpu] Reimplement XLA_Shape in C API for performance.
|
2020-10-19 14:17:22 -07:00 |
|
c_api_defn.h
|
Create dummy topology types for libtpu C API.
|
2020-08-10 17:13:50 -07:00 |
|
noncopyable_buffer.h
|
Add TPU Transfer Manager Interface and Noncopyable Buffer support
|
2020-08-05 15:22:36 -07:00 |
|
proto_helper.cc
|
|
|
|
proto_helper.h
|
Implement Serialization/Deserialization for remote compilation cache.
|
2020-09-01 12:51:30 -07:00 |
|
status_helper.h
|
Make StatusHelper safer to use and fix a related memory leak.
|
2020-07-24 21:44:55 -07:00 |
|
tpu_computation_placer.cc
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_computation_placer.h
|
Add TpuComputationPlacer::AssignLocalDevices() to TPU API.
|
2020-09-03 11:49:36 -07:00 |
|
tpu_event.h
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_executable_interface.cc
|
[TPU] Don't pass host_shapes across the TPU API boundary. They can be computed from device shapes.
|
2020-10-07 18:48:10 -07:00 |
|
tpu_executable_interface.h
|
Add TpuExecutableInterface::fingerprint() virtual method.
|
2020-09-09 18:41:01 -07:00 |
|
tpu_executable.cc
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_executable.h
|
Add TpuExecutableInterface::fingerprint() virtual method.
|
2020-09-09 18:41:01 -07:00 |
|
tpu_executor_c_api.h
|
Add HostCount() / ChipsPerHost() API to TPUTopologyExternal.
|
2020-10-07 15:39:48 -07:00 |
|
tpu_executor_interface.h
|
Add TpuExecutorInterface::GetCoreLocationExternal() to TPU API.
|
2020-08-17 13:15:15 -07:00 |
|
tpu_executor.cc
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_executor.h
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_node_context_c_api.h
|
Remove two unused LOG(FATAL) methods from TpuNodeContext.
|
2020-07-22 20:26:40 -07:00 |
|
tpu_node_context.cc
|
Fix some API linking issues within TPU 1VM (some API calls could only be statically linked)
|
2020-07-23 23:21:47 -07:00 |
|
tpu_node_context.h
|
Remove two unused LOG(FATAL) methods from TpuNodeContext.
|
2020-07-22 20:26:40 -07:00 |
|
tpu_platform_interface.cc
|
Refactor TpuPlatformInterface::GetRegisteredPlatform() to make tries_left configurable.
|
2020-09-21 19:53:06 -07:00 |
|
tpu_platform_interface.h
|
[XLA:PJRT] Add TPU client and associated libtpu.so initializer.
|
2020-10-06 15:15:22 -07:00 |
|
tpu_platform_registration.cc
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_platform.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_platform.h
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_stream_interface.h
|
Clean up build files.
|
2020-06-25 17:32:38 -07:00 |
|
tpu_stream.h
|
Add TpuStream::EnqueueTransferHostToDevice and TpuStream::EnqueueTransferDeviceToHost
|
2020-09-08 16:22:00 -07:00 |
|
tpu_timer.h
|
Rename tpu_load_library to tpu_api_dlsym_initializer.
|
2020-06-22 17:45:58 -07:00 |
|
tpu_topology.cc
|
Add HostCount() / ChipsPerHost() API to TPUTopologyExternal.
|
2020-10-07 15:39:48 -07:00 |
|
tpu_topology.h
|
Add HostCount() / ChipsPerHost() API to TPUTopologyExternal.
|
2020-10-07 15:39:48 -07:00 |
|
tpu_transfer_manager_interface.cc
|
Introduce additional TPU infeed and outfeed ops
|
2020-08-07 17:57:57 -07:00 |
|
tpu_transfer_manager_interface.h
|
Introduce additional TPU infeed and outfeed ops
|
2020-08-07 17:57:57 -07:00 |
|
tpu_transfer_manager_registration.cc
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_transfer_manager.cc
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
|
tpu_transfer_manager.h
|
Put a bunch of TPU classes in the tensorflow::tpu namespace.
|
2020-09-21 13:59:05 -07:00 |
