| .. |
|
cuda
|
Merge pull request #46300 from nluehr:tf32-fixes
|
2021-02-08 14:08:55 -08:00 |
|
gpu
|
Use device handle instead of gpu ordinal in GpuVirtualMemAllocator for
|
2021-01-21 16:11:33 -08:00 |
|
host
|
Add target environment constraints to a subset of TensorFlow packages and targets.
|
2020-09-21 10:57:01 -07:00 |
|
lib
|
internal tests cleanup
|
2020-11-03 19:22:55 -08:00 |
|
platform
|
Make TryDlopenCUDALibraries use the cached SO loaders
|
2021-01-05 15:13:31 -08:00 |
|
rocm
|
[ROCm] Fix for a bug in ROCm batchnorm implementation.
|
2021-01-22 17:33:54 +00:00 |
|
tpu
|
Add a TpuTopologyExternal::CoreForId method.
|
2021-01-29 12:05:55 -08:00 |
|
allocator_stats.cc
|
|
|
|
allocator_stats.h
|
|
|
|
blas.cc
|
Qualifying "string" as std::string
|
2021-02-05 16:16:25 -08:00 |
|
blas.h
|
Qualifying "string" as std::string
|
2021-02-05 16:16:25 -08:00 |
|
BUILD
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07:00 |
|
build_defs.bzl
|
Remove cuda_is_configured() non-configurable macro.
|
2021-01-21 10:37:14 -08:00 |
|
data_type.h
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07:00 |
|
device_description.cc
|
Fix a few ClangTidy warnings
|
2021-01-12 06:02:22 -08:00 |
|
device_description.h
|
Merge pull request #46232 from ROCmSoftwarePlatform:google_upstream_rocm_use_gcn_arch_name
|
2021-01-12 04:11:36 -08:00 |
|
device_memory_allocator.h
|
Add DCHECK to ScopedDeviceMemory constructor.
|
2020-09-23 10:24:28 -07:00 |
|
device_memory.h
|
|
|
|
device_options.h
|
|
|
|
dnn.cc
|
add missing ActivationMode::kNone to ActivationModeString
|
2020-12-27 19:45:23 -08:00 |
|
dnn.h
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07:00 |
|
dnn.proto
|
Use (+generalize) existing dnn::DataType in blas::
|
2020-09-29 16:07:57 +10:00 |
|
event.cc
|
|
|
|
event.h
|
|
|
|
executor_cache.cc
|
|
|
|
executor_cache.h
|
|
|
|
fft.h
|
|
|
|
gpu_launch_dim.h
|
|
|
|
host_or_device_scalar.h
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07:00 |
|
kernel_cache_config.h
|
|
|
|
kernel_spec.cc
|
|
|
|
kernel_spec.h
|
|
|
|
kernel.cc
|
|
|
|
kernel.h
|
|
|
|
launch_dim.h
|
|
|
|
module_spec.h
|
|
|
|
multi_platform_manager.cc
|
Open source distributed_tpu_rewrite_pass.cc and associated helper methods
|
2020-07-21 16:12:43 -07:00 |
|
multi_platform_manager.h
|
Open source distributed_tpu_rewrite_pass.cc and associated helper methods
|
2020-07-21 16:12:43 -07:00 |
|
platform.cc
|
|
|
|
platform.h
|
|
|
|
plugin_registry.cc
|
Qualify uses of std::string
|
2020-07-07 18:25:15 -07:00 |
|
plugin_registry.h
|
|
|
|
plugin.cc
|
|
|
|
plugin.h
|
|
|
|
rng.cc
|
|
|
|
rng.h
|
|
|
|
scratch_allocator.cc
|
|
|
|
scratch_allocator.h
|
|
|
|
stream_executor_internal.cc
|
|
|
|
stream_executor_internal.h
|
Remove SharedMemoryConfig since it is not used anywhere.
|
2020-08-11 20:01:04 -07:00 |
|
stream_executor_pimpl.cc
|
Change Device::Sync to explicitly sync with all GPU streams
|
2020-10-26 09:43:11 -07:00 |
|
stream_executor_pimpl.h
|
Change Device::Sync to explicitly sync with all GPU streams
|
2020-10-26 09:43:11 -07:00 |
|
stream_executor.h
|
|
|
|
stream_test.cc
|
|
|
|
stream.cc
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07:00 |
|
stream.h
|
Fix typo in comment
|
2021-01-27 13:01:01 -08:00 |
|
temporary_device_memory.cc
|
|
|
|
temporary_device_memory.h
|
|
|
|
temporary_memory_manager.cc
|
|
|
|
temporary_memory_manager.h
|
|
|
|
tf_allocator_adapter.cc
|
s/no_retry_on_failure/retry_on_failure; NFC
|
2020-08-06 12:01:41 -07:00 |
|
tf_allocator_adapter.h
|
|
|
|
timer.cc
|
|
|
|
timer.h
|
|
|
|
trace_listener.h
|
|
|
