| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | BUILD | 8 "//tensorflow/stream_executor:build_defs.bzl",
41 "//tensorflow/stream_executor/gpu:gpu_diagnostics_header",
42 "//tensorflow/stream_executor/lib",
43 "//tensorflow/stream_executor/platform",
56 "//tensorflow/stream_executor:device_options",
57 "//tensorflow/stream_executor/gpu:gpu_driver_header",
58 "//tensorflow/stream_executor/lib",
59 "//tensorflow/stream_executor/platform",
60 "//tensorflow/stream_executor/platform:dso_loader",
72 "//tensorflow/stream_executor",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 8 "//tensorflow/stream_executor:build_defs.bzl",
40 deps = ["//tensorflow/stream_executor:platform"],
53 "//tensorflow/stream_executor", # buildcleaner: keep
54 "//tensorflow/stream_executor:executor_cache",
55 "//tensorflow/stream_executor:multi_platform_manager",
56 "//tensorflow/stream_executor:stream_executor_pimpl_header",
57 "//tensorflow/stream_executor/lib",
58 "//tensorflow/stream_executor/platform",
71 "//tensorflow/stream_executor/gpu:gpu_diagnostics_header",
72 "//tensorflow/stream_executor/lib",
[all …]
|
| D | cufft_stub.cc | 28 auto handle_or = stream_executor::internal::DsoLoader::GetCufftDsoHandle(); in GetDsoHandle()
40 stream_executor::port::Env::Default() in LoadSymbol()
|
| D | curand_stub.cc | 28 auto handle_or = stream_executor::internal::DsoLoader::GetCurandDsoHandle(); in GetDsoHandle()
40 stream_executor::port::Env::Default() in LoadSymbol()
|
| D | cublas_stub.cc | 30 auto handle_or = stream_executor::internal::DsoLoader::GetCublasDsoHandle(); in GetDsoHandle()
42 stream_executor::port::Env::Default() in LoadSymbol()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 7 "//tensorflow/stream_executor:build_defs.bzl",
14 default_visibility = ["//tensorflow/stream_executor:__subpackages__"],
29 deps = ["//tensorflow/stream_executor/platform"],
39 "//tensorflow/stream_executor",
40 "//tensorflow/stream_executor:stream_executor_internal",
41 "//tensorflow/stream_executor/platform",
49 "//tensorflow/stream_executor/lib",
50 "//tensorflow/stream_executor/platform",
59 "//tensorflow/stream_executor:device_options",
60 "//tensorflow/stream_executor/lib",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | BUILD | 12 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
39 "//tensorflow/stream_executor/platform",
50 "//tensorflow/stream_executor/lib",
51 "//tensorflow/stream_executor/platform",
92 "//tensorflow/stream_executor/lib",
93 "//tensorflow/stream_executor/platform",
126 "stream_executor.h",
139 "//tensorflow/stream_executor/lib",
140 "//tensorflow/stream_executor/platform",
153 "//tensorflow/stream_executor/platform",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | BUILD | 6 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
33 "//tensorflow/stream_executor:platform",
49 "//tensorflow/stream_executor:executor_cache",
50 "//tensorflow/stream_executor:multi_platform_manager",
51 "//tensorflow/stream_executor:stream_executor_headers",
52 "//tensorflow/stream_executor/lib",
53 "//tensorflow/stream_executor/platform",
67 "//tensorflow/stream_executor:kernel",
68 "//tensorflow/stream_executor/lib",
81 "//tensorflow/stream_executor:stream_executor_internal",
[all …]
|
| /external/tensorflow/tensorflow/core/platform/default/build_config/ |
| D | BUILD | 36 name = "stream_executor",
38 "//tensorflow/stream_executor/cuda:cuda_activation",
39 "//tensorflow/stream_executor/rocm:rocm_activation",
42 "//tensorflow/stream_executor",
43 "//tensorflow/stream_executor:dnn",
44 "//tensorflow/stream_executor:event",
45 "//tensorflow/stream_executor:multi_platform_manager",
46 "//tensorflow/stream_executor:scratch_allocator",
47 "//tensorflow/stream_executor/cuda:cuda_platform_id",
48 "//tensorflow/stream_executor/host:host_platform_id",
[all …]
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_stream_executor.cmake | 16 # name = "stream_executor",
56 "${tensorflow_source_dir}/tensorflow/stream_executor/*.cc"
57 "${tensorflow_source_dir}/tensorflow/stream_executor/*.h"
58 "${tensorflow_source_dir}/tensorflow/stream_executor/lib/*.cc"
59 "${tensorflow_source_dir}/tensorflow/stream_executor/lib/*.h"
60 "${tensorflow_source_dir}/tensorflow/stream_executor/platform/*.h"
61 "${tensorflow_source_dir}/tensorflow/stream_executor/platform/default/*.h"
66 "${tensorflow_source_dir}/tensorflow/stream_executor/cuda/*.cc"
70 "${tensorflow_source_dir}/tensorflow/stream_executor/cuda/*_test.cc"
78 "${tensorflow_source_dir}/tensorflow/stream_executor/*test.cc"
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.h | 23 namespace stream_executor {
56 ExecutableRunOptions& set_stream(stream_executor::Stream* stream);
57 stream_executor::Stream* stream() const;
63 stream_executor::Stream* stream);
64 stream_executor::Stream* host_to_device_stream() const;
87 stream_executor::Stream* stream_ = nullptr;
91 stream_executor::Stream* host_to_device_stream_ = nullptr;
|
| D | executable_run_options.cc | 39 stream_executor::Stream* stream) { in set_stream()
44 stream_executor::Stream* ExecutableRunOptions::stream() const { in stream()
49 stream_executor::Stream* stream) { in set_host_to_device_stream()
54 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream()
|
| /external/tensorflow/tensorflow/stream_executor/platform/default/ |
| D | BUILD | 3 package(default_visibility = ["//tensorflow/stream_executor:__subpackages__"])
19 "//tensorflow/stream_executor:platform",
20 "//tensorflow/stream_executor/lib",
21 "//tensorflow/stream_executor/platform",
|
| D | initialize.h | 23 namespace stream_executor {
49 ::stream_executor::port::Initializer google_initializer_##type##_##name( \
56 extern ::stream_executor::port::Initializer google_initializer_##type##_##name
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | device_memory_allocator.cc | 35 TF_ASSIGN_OR_RETURN(se::StreamExecutor * stream_executor, in Allocate()
37 se::DeviceMemoryBase result = stream_executor->AllocateArray<uint8>(size); in Allocate()
53 TF_ASSIGN_OR_RETURN(se::StreamExecutor * stream_executor, in Deallocate()
57 stream_executor->Deallocate(&mem); in Deallocate()
|
| D | backend.cc | 131 TF_ASSIGN_OR_RETURN(auto executor, stream_executor(device_ordinal)); in BorrowStream()
193 StatusOr<se::StreamExecutor*> Backend::stream_executor( in stream_executor() function in xla::Backend
217 stream_executor(device_ordinal_a)); in devices_equivalent()
219 stream_executor(device_ordinal_b)); in devices_equivalent()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 38 namespace stream_executor {
71 virtual stream_executor::Stream* stream() const { return nullptr; } in stream()
73 stream_executor::Stream* stream) const { in MaintainLifetimeOnStream()
105 virtual Status ThenExecute(Device* device, stream_executor::Stream* stream, in ThenExecute()
152 stream_executor::Stream* stream = nullptr;
|
| /external/tensorflow/tensorflow/stream_executor/lib/ |
| D | status.h | 26 namespace stream_executor {
33 ASSERT_EQ(::stream_executor::port::Status::OK(), (val))
55 using namespace stream_executor; // NOLINT[build/namespaces] variable
|
| D | ptr_util.h | 22 namespace stream_executor {
33 using namespace stream_executor; // NOLINT[build/namespaces] variable
|
| D | array_slice.h | 21 namespace stream_executor {
38 using namespace stream_executor; // NOLINT[build/namespaces] variable
|
| /external/tensorflow/tensorflow/stream_executor/platform/ |
| D | BUILD | 3 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
27 "//tensorflow/stream_executor/platform/default:platform",
37 "//tensorflow/stream_executor/platform/default:dso_loader",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 110 stream_executor::StreamExecutor* executor);
143 std::map<stream_executor::StreamExecutor*, se::ScopedModuleHandle>
145 std::map<stream_executor::StreamExecutor*, BufferAllocToDeviceMemoryMap>
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 34 stream_executor::StreamExecutor* stream_executor) in XlaDeviceAllocator() argument
35 : stream_executor_(stream_executor) {} in XlaDeviceAllocator()
56 absl::optional<stream_executor::AllocatorStats> se_stats = in GetStats()
220 stream_executor::port::StatusOr<xla::StreamPool::Ptr> ptr_or_status = in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_init.h | 22 namespace stream_executor {
36 stream_executor::Platform* GPUMachineManager();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 31 tensorflow::CudnnVersion GetCudnnVersion(se::StreamExecutor* stream_executor) { in GetCudnnVersion() argument
33 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion()
46 se::StreamExecutor* stream_executor) { in GetComputeCapability() argument
49 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
|