|
..
|
|
cuda
|
|
gpu
|
|
host
|
|
lib
|
|
platform
|
|
rocm
|
|
tpu
|
|
BUILD
|
|
allocator_stats.cc
|
|
allocator_stats.h
|
|
blas.cc
|
|
blas.h
|
|
build_defs.bzl
|
|
data_type.h
|
|
device_description.cc
|
|
device_description.h
|
|
device_memory.h
|
|
device_memory_allocator.h
|
|
device_options.h
|
|
dnn.cc
|
|
dnn.h
|
|
dnn.proto
|
|
event.cc
|
|
event.h
|
|
executor_cache.cc
|
|
executor_cache.h
|
|
fft.h
|
|
gpu_launch_dim.h
|
|
host_or_device_scalar.h
|
|
kernel.cc
|
|
kernel.h
|
|
kernel_cache_config.h
|
|
kernel_spec.cc
|
|
kernel_spec.h
|
|
launch_dim.h
|
|
lazy_op_runner.h
|
|
matmul_util.cc
|
|
matmul_util.h
|
|
module_spec.h
|
|
multi_platform_manager.cc
|
|
multi_platform_manager.h
|
|
platform.cc
|
|
platform.h
|
|
plugin.cc
|
|
plugin.h
|
|
plugin_registry.cc
|
|
plugin_registry.h
|
|
rng.cc
|
|
rng.h
|
|
scratch_allocator.cc
|
|
scratch_allocator.h
|
|
stream.cc
|
|
stream.h
|
|
stream_executor.h
|
|
stream_executor_internal.cc
|
|
stream_executor_internal.h
|
|
stream_executor_pimpl.cc
|
|
stream_executor_pimpl.h
|
|
stream_test.cc
|
|
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
|