| .. |
|
cuda
|
Switch from //tensorflow:using_cuda_{clang,nvcc} to //tensorflow:is_cuda_enabled.
|
2021-03-03 08:47:26 -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
|
Merge pull request #46543 from kaixih:cudnn_rnn_update_tf2_pr
|
2021-03-01 22:10:03 -08:00 |
|
tpu
|
Introduce TPUReshardVariable op to open source TensorFlow
|
2021-02-23 11:24:15 -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
|
Upgrade to CUDNN RNN v8 APIs
|
2021-02-18 15:03:56 -08: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
|
|
|
|
multi_platform_manager.h
|
|
|
|
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
|
|
|
|
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
|
Upgrade to CUDNN RNN v8 APIs
|
2021-02-18 15:03:56 -08:00 |
|
stream.h
|
Merge pull request #46543 from kaixih:cudnn_rnn_update_tf2_pr
|
2021-03-01 22:10:03 -08:00 |
|
temporary_device_memory.cc
|
|
|
|
temporary_device_memory.h
|
|
|
|
temporary_memory_manager.cc
|
|
|
|
temporary_memory_manager.h
|
|
|
|
tf_allocator_adapter.cc
|
|
|
|
tf_allocator_adapter.h
|
|
|
|
timer.cc
|
|
|
|
timer.h
|
|
|
|
trace_listener.h
|
|
|
