| .. |
|
cuda
|
Lower priority of missing NUMA /sys file log message.
|
2021-01-12 11:51:14 -08:00 |
|
gpu
|
Adding hooks in the Stream Executor API to get/set the AMDGPU gcnArchName device property
|
2021-01-11 22:57:46 +00: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
|
Fix a few ClangTidy warnings
|
2021-01-12 06:02:22 -08:00 |
|
tpu
|
Add TPU runtime version.
|
2021-01-14 13:29:47 -08:00 |
|
allocator_stats.cc
|
|
|
|
allocator_stats.h
|
|
|
|
blas.cc
|
Use (+generalize) existing dnn::DataType in blas::
|
2020-09-29 16:07:57 +10:00 |
|
blas.h
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07:00 |
|
BUILD
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07:00 |
|
build_defs.bzl
|
|
|
|
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
|
|
|
|
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
|
Merge pull request #43237 from benbarsdell:cublaslt
|
2020-10-15 14:39:38 -07: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
|
|
|
