STT-tensorflow/tensorflow/stream_executor
Mihai Maruseac b550171e78 Prevent CHECK-fail in LSTM/GRU with zero-length input.
PiperOrigin-RevId: 346239181
Change-Id: I5f233dbc076aab7bb4e31ba24f5abd4eaf99ea4f
2020-12-08 17:41:20 -08:00
..
cuda Prevent CHECK-fail in LSTM/GRU with zero-length input. 2020-12-08 17:41:20 -08:00
gpu Merge pull request #43888 from nouiz:upstream_old_ptxas 2020-10-16 10:42:20 -07:00
host Add target environment constraints to a subset of TensorFlow packages and targets. 2020-09-21 10:57:01 -07:00
lib Add target environment constraints to a subset of TensorFlow packages and targets. 2020-09-21 10:57:01 -07:00
platform Merge pull request #43237 from benbarsdell:cublaslt 2020-10-15 14:39:38 -07:00
rocm Populating memory bandwidth information in the DeviceDescription, using information from the call to hipGetDeviceProperties 2020-12-04 19:24:47 +00:00
tpu [libtpu] Fix static_asserts. 2020-10-19 21:34:13 -07:00
allocator_stats.cc Export memory fragmentation in memstats. 2020-04-14 22:18:56 -07:00
allocator_stats.h Export memory fragmentation in memstats. 2020-04-14 22:18:56 -07:00
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 use static linked cupti internally. 2020-05-29 13:47:38 -07:00
data_type.h Merge pull request #43237 from benbarsdell:cublaslt 2020-10-15 14:39:38 -07:00
device_description.cc Merge pull request #41757 from tg-at-google:wsign-compare-semi-final-lite-python-stream-executor 2020-07-28 12:59:56 -07:00
device_description.h Qualify uses of std::string 2020-02-25 15:07:45 -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 [XLA:Python] Specify a 2MiB stack size for host stream threads. 2020-05-01 14:20:58 -07:00
dnn.cc PR #37679: Support two CUDNN CTC Loss algorithms 2020-04-10 14:15:07 -07: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 Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
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 Qualify uses of std::string 2020-02-25 15:07:45 -08:00
kernel_spec.h Qualify uses of std::string 2020-02-25 15:07:45 -08:00
kernel.cc Qualify uses of std::string 2020-02-25 15:07:45 -08:00
kernel.h Avoid a footgun in KernelArgsArray 2020-03-02 11:50:16 -08:00
launch_dim.h Qualify uses of std::string 2020-02-25 15:07:45 -08:00
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 Qualify uses of std::string 2020-02-25 15:07:45 -08:00
platform.h Qualify uses of std::string 2020-02-25 15:07:45 -08:00
plugin_registry.cc Qualify uses of std::string 2020-07-07 18:25:15 -07:00
plugin_registry.h Qualify uses of std::string 2020-02-25 15:07:45 -08:00
plugin.cc
plugin.h
rng.cc
rng.h Some const declarations changed to constexpr 2020-04-21 10:21:10 -07:00
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 Merge pull request #43237 from benbarsdell:cublaslt 2020-10-15 14:39:38 -07:00
stream_executor_pimpl.h Merge pull request #43237 from benbarsdell:cublaslt 2020-10-15 14:39:38 -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 Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
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