| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | BUILD | 6 "//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/compiler/xla/stream_executor/cuda/ |
| D | BUILD | 7 "//tensorflow/stream_executor:build_defs.bzl",
43 deps = ["//tensorflow/compiler/xla/stream_executor:platform"],
57 "//tensorflow/compiler/xla/stream_executor", # buildcleaner: keep
58 "//tensorflow/compiler/xla/stream_executor:executor_cache",
59 "//tensorflow/compiler/xla/stream_executor:multi_platform_manager",
60 "//tensorflow/compiler/xla/stream_executor:stream_executor_pimpl_header",
61 "//tensorflow/compiler/xla/stream_executor/lib",
62 "//tensorflow/compiler/xla/stream_executor/platform",
82 "//tensorflow/compiler/xla/stream_executor/gpu:gpu_diagnostics_header",
83 "//tensorflow/compiler/xla/stream_executor/lib",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| D | BUILD | 5 "//tensorflow/stream_executor:build_defs.bzl",
23 "//tensorflow/compiler/xla/stream_executor:__subpackages__",
25 "//tensorflow/stream_executor:__subpackages__",
33 deps = ["//tensorflow/compiler/xla/stream_executor/platform"],
44 "//tensorflow/compiler/xla/stream_executor",
45 "//tensorflow/compiler/xla/stream_executor:stream_executor_internal",
46 "//tensorflow/compiler/xla/stream_executor/platform",
54 "//tensorflow/compiler/xla/stream_executor/lib",
55 "//tensorflow/compiler/xla/stream_executor/platform",
64 "//tensorflow/compiler/xla/stream_executor:__subpackages__",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | BUILD | 4 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
22 "//tensorflow/compiler/xla/stream_executor/host:host_platform_id",
23 "//tensorflow/stream_executor:platform",
36 "//tensorflow/compiler/xla/stream_executor/host:host_platform",
37 "//tensorflow/stream_executor:executor_cache",
38 "//tensorflow/stream_executor:multi_platform_manager",
39 "//tensorflow/stream_executor:stream_executor_headers",
40 "//tensorflow/stream_executor/lib",
41 "//tensorflow/stream_executor/platform",
55 "//tensorflow/compiler/xla/stream_executor/host:host_stream",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | BUILD | 45 "stream_executor.h",
58 "//tensorflow/compiler/xla/stream_executor/lib",
59 "//tensorflow/compiler/xla/stream_executor/platform",
77 "//tensorflow/compiler/xla/stream_executor/platform",
90 "//tensorflow/compiler/xla/stream_executor/lib",
91 "//tensorflow/compiler/xla/stream_executor/platform",
102 "//tensorflow/compiler/xla/stream_executor/lib",
103 "//tensorflow/compiler/xla/stream_executor/platform",
114 "//tensorflow/compiler/xla/stream_executor/platform",
129 "//tensorflow/compiler/xla/stream_executor/lib",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 5 "//tensorflow/stream_executor:build_defs.bzl",
18 "//tensorflow/compiler/xla/stream_executor:__subpackages__",
19 "//tensorflow/stream_executor:__subpackages__",
27 deps = ["//tensorflow/compiler/xla/stream_executor/gpu:gpu_activation_header"],
34 "//tensorflow/compiler/xla/stream_executor/gpu:gpu_activation",
42 "//tensorflow/compiler/xla/stream_executor/gpu:gpu_diagnostics_header",
51 "//tensorflow/compiler/xla/stream_executor:__subpackages__",
53 "//tensorflow/stream_executor:__subpackages__",
56 "//tensorflow/compiler/xla/stream_executor/gpu:gpu_driver_header",
64 "//tensorflow/compiler/xla/stream_executor/gpu:gpu_event_header",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_platform.h | 36 absl::flat_hash_map<stream_executor::internal::StreamInterface*,
39 absl::flat_hash_map<stream_executor::internal::EventInterface*,
42 static const ::stream_executor::Platform::Id kId;
44 using Status = ::stream_executor::port::Status;
46 using StatusOr = ::stream_executor::port::StatusOr<T>;
80 StatusOr<std::unique_ptr<::stream_executor::DeviceDescription>>
85 StatusOr<::stream_executor::StreamExecutor*> ExecutorForDevice( in ExecutorForDevice()
87 stream_executor::StreamExecutorConfig config; in ExecutorForDevice()
92 StatusOr<::stream_executor::StreamExecutor*>
95 const ::stream_executor::PluginConfig& plugin_config) override { in ExecutorForDeviceWithPluginConfig()
[all …]
|
| D | tpu_executor.h | 44 using Status = ::stream_executor::port::Status;
46 using StatusOr = ::stream_executor::port::StatusOr<T>;
48 using Stream = ::stream_executor::Stream;
49 using Event = ::stream_executor::Event;
50 using Timer = ::stream_executor::Timer;
51 using DeviceMemoryBase = ::stream_executor::DeviceMemoryBase;
52 using StreamInterface = ::stream_executor::internal::StreamInterface;
54 ::stream_executor::internal::StreamExecutorInterface;
57 absl::flat_hash_map<stream_executor::internal::TimerInterface*,
67 ::stream_executor::DeviceOptions device_options) override;
[all …]
|
| D | c_api_defn.h | 29 namespace stream_executor {
35 stream_executor::Platform* platform;
39 stream_executor::StreamExecutor* executor;
43 explicit SE_Stream(stream_executor::StreamExecutor* parent) in SE_Stream()
45 stream_executor::Stream stream;
49 explicit SE_Event(stream_executor::StreamExecutor* parent) : event(parent) {} in SE_Event()
50 stream_executor::Event event;
54 explicit SE_Timer(stream_executor::StreamExecutor* parent) : timer(parent) {} in SE_Timer()
55 stream_executor::Timer timer;
59 stream_executor::StreamExecutorConfig config;
[all …]
|
| D | tpu_transfer_manager.h | 35 using Status = stream_executor::port::Status;
37 using StatusOr = stream_executor::port::StatusOr<T>;
39 stream_executor::Platform::Id PlatformId() const override;
45 stream_executor::Stream* stream, const xla::LiteralSlice& literal,
50 stream_executor::Stream* stream, const xla::ShapedBuffer& device_buffer,
54 Status TransferLiteralToInfeed(stream_executor::StreamExecutor* executor,
58 stream_executor::StreamExecutor* executor,
66 absl::Span<stream_executor::StreamExecutor* const> executor) override;
74 stream_executor::StreamExecutor* executor,
82 stream_executor::Stream* stream,
[all …]
|
| D | tpu_platform.cc | 28 const ::stream_executor::Platform::Id TpuPlatform::kId = GetTpuPlatformId();
31 using Status = ::stream_executor::port::Status;
33 using StatusOr = ::stream_executor::port::StatusOr<T>;
82 StatusOr<::stream_executor::StreamExecutor*> TpuPlatform::GetExecutor( in GetExecutor()
83 const ::stream_executor::StreamExecutorConfig& config) { in GetExecutor()
88 StatusOr<std::unique_ptr<::stream_executor::StreamExecutor>>
90 const ::stream_executor::StreamExecutorConfig& config) { in GetUncachedExecutor()
104 return std::make_unique<stream_executor::StreamExecutor>( in GetUncachedExecutor()
108 ::stream_executor::Platform::Id TpuPlatform::id() const { in id()
137 void TpuPlatform::InsertEvent(stream_executor::internal::EventInterface* key, in InsertEvent()
[all …]
|
| D | tpu_executor.cc | 26 using stream_executor::DeviceMemoryBase;
32 using ::stream_executor::port::Status;
40 ::stream_executor::DeviceOptions device_options) { in Init()
120 bool TpuExecutor::StartTimer(Stream* stream, ::stream_executor::Timer* timer) { in StartTimer()
126 bool TpuExecutor::StopTimer(Stream* stream, ::stream_executor::Timer* timer) { in StopTimer()
132 stream_executor::Event::Status TpuExecutor::PollForEventStatus( in PollForEventStatus()
133 stream_executor::Event* event) { in PollForEventStatus()
135 return stream_executor::Event::Status( in PollForEventStatus()
141 ::stream_executor::Event* event) { in RecordEvent()
151 ::stream_executor::Event* event) { in WaitForEvent()
[all …]
|
| D | BUILD | 10 "//tensorflow/stream_executor:__subpackages__",
23 "//tensorflow/compiler/xla/stream_executor:stream_executor_headers",
56 "//tensorflow/compiler/xla/stream_executor:device_memory",
57 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
118 "//tensorflow/compiler/xla/stream_executor",
119 "//tensorflow/compiler/xla/stream_executor/lib",
120 "//tensorflow/compiler/xla/stream_executor/platform",
136 "//tensorflow/compiler/xla/stream_executor:stream_executor_headers",
161 "//tensorflow/compiler/xla/stream_executor:stream_executor_headers",
162 "//tensorflow/compiler/xla/stream_executor/lib",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | BUILD | 5 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
27 "//tensorflow/compiler/xla/stream_executor:platform",
43 "//tensorflow/compiler/xla/stream_executor:executor_cache",
44 "//tensorflow/compiler/xla/stream_executor:multi_platform_manager",
45 "//tensorflow/compiler/xla/stream_executor:stream_executor_headers",
46 "//tensorflow/compiler/xla/stream_executor/lib",
47 "//tensorflow/compiler/xla/stream_executor/platform",
64 "//tensorflow/compiler/xla/stream_executor:kernel",
65 "//tensorflow/compiler/xla/stream_executor/lib",
80 "//tensorflow/compiler/xla/stream_executor:stream_executor_internal",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | BUILD | 10 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
29 "//tensorflow/compiler/xla/stream_executor:launch_dim",
37 "//tensorflow/compiler/xla/stream_executor:device_description",
53 "//tensorflow/compiler/xla/stream_executor:event",
70 "stream_executor.h",
76 "//tensorflow/compiler/xla/stream_executor:kernel",
84 "//tensorflow/compiler/xla/stream_executor:kernel_spec",
92 "//tensorflow/compiler/xla/stream_executor:kernel_cache_config",
100 "//tensorflow/compiler/xla/stream_executor:module_spec",
112 actual = ":stream_executor",
[all …]
|
| /external/tensorflow/tensorflow/tsl/platform/default/build_config/ |
| D | BUILD | 39 "//conditions:default": "//tensorflow/stream_executor/cuda:all_runtime",
45 name = "stream_executor",
47 "//tensorflow/stream_executor/cuda:cuda_activation",
48 "//tensorflow/stream_executor/rocm:rocm_activation",
51 "//tensorflow/stream_executor",
52 "//tensorflow/stream_executor:dnn",
53 "//tensorflow/stream_executor:event",
54 "//tensorflow/stream_executor:multi_platform_manager",
55 "//tensorflow/stream_executor:scratch_allocator",
56 "//tensorflow/stream_executor/cuda:cuda_platform_id",
[all …]
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | BUILD | 20 "stream_executor.h",
27 hdrs = ["stream_executor.h"],
36 name = "stream_executor",
37 srcs = ["stream_executor.cc"],
38 hdrs = ["stream_executor.h"],
49 "//tensorflow/stream_executor:executor_cache",
50 "//tensorflow/stream_executor:multi_platform_manager",
51 "//tensorflow/stream_executor:platform",
52 "//tensorflow/stream_executor:stream_executor_internal",
53 "//tensorflow/stream_executor:stream_executor_pimpl",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 6 "//tensorflow/stream_executor:build_defs.bzl",
37 deps = ["//tensorflow/compiler/xla/stream_executor/cuda:cuda_platform_id"],
46 "//tensorflow/compiler/xla/stream_executor/cuda:cuda_platform",
55 "//tensorflow/compiler/xla/stream_executor/cuda:cuda_diagnostics",
62 "//tensorflow/compiler/xla/stream_executor/cuda:cuda_stub",
78 "//tensorflow/compiler/xla/stream_executor/cuda:cuda_driver",
86 "//tensorflow/compiler/xla/stream_executor/cuda:cudart_stub",
97 "//tensorflow/compiler/xla/stream_executor/cuda:cuda_activation_header",
105 "//tensorflow/compiler/xla/stream_executor/cuda:cuda_activation",
114 "//tensorflow/compiler/xla/stream_executor/cuda:cuda_gpu_executor_header",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | BUILD | 29 "//tensorflow/compiler/xla/stream_executor/tpu:c_api_decl",
38 "//tensorflow/compiler/xla/stream_executor/tpu:tpu_executor_c_api_hdrs",
46 "//tensorflow/compiler/xla/stream_executor/tpu:c_api_conversions",
55 "//tensorflow/compiler/xla/stream_executor/tpu:noncopyable_buffer",
63 "//tensorflow/compiler/xla/stream_executor/tpu:status_helper",
71 "//tensorflow/compiler/xla/stream_executor/tpu:proto_helper",
85 "//tensorflow/compiler/xla/stream_executor/tpu:tpu_executor",
94 "//tensorflow/compiler/xla/stream_executor/tpu:tpu_platform_hdr",
110 "//tensorflow/compiler/xla/stream_executor/tpu:tpu_executor_hdrs",
118 "//tensorflow/compiler/xla/stream_executor/tpu:tpu_platform_id",
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_virtual_mem_allocator.h | 43 static stream_executor::port::StatusOr<
47 stream_executor::gpu::GpuContext& gpu_context, PlatformDeviceId gpu_id,
77 stream_executor::gpu::GpuContext& gpu_context, PlatformDeviceId gpu_id,
78 std::vector<stream_executor::gpu::GpuDeviceHandle> access_device_handles,
79 stream_executor::gpu::GpuDriver::VmemSpan vmem, size_t granularity);
81 stream_executor::gpu::GpuContext& gpu_context_;
88 const std::vector<stream_executor::gpu::GpuDeviceHandle> access_gpu_handles_;
91 stream_executor::gpu::GpuDriver::VmemSpan vmem_;
101 stream_executor::gpu::GpuDevicePtr va;
102 stream_executor::gpu::GpuDriver::GenericMemoryHandle physical;
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.h | 27 namespace stream_executor {
79 std::function<void(stream_executor::Stream*, std::function<void()>)>;
86 stream_executor::DeviceMemoryAllocator* allocator);
87 stream_executor::DeviceMemoryAllocator* allocator() const;
100 ExecutableRunOptions& set_stream(stream_executor::Stream* stream);
101 stream_executor::Stream* stream() const;
107 stream_executor::Stream* stream);
108 stream_executor::Stream* host_to_device_stream() const;
159 stream_executor::DeviceMemoryAllocator* allocator_ = nullptr;
162 stream_executor::Stream* stream_ = nullptr;
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | BUILD | 44 "//tensorflow/c/experimental/stream_executor",
45 "//tensorflow/c/experimental/stream_executor:stream_executor_internal",
55 "//tensorflow/core/platform:stream_executor",
56 "//tensorflow/stream_executor:event",
57 "//tensorflow/stream_executor:kernel",
77 "//tensorflow/c/experimental/stream_executor",
78 "//tensorflow/c/experimental/stream_executor:stream_executor_internal",
89 "//tensorflow/core/platform:stream_executor",
90 "//tensorflow/stream_executor:event",
91 "//tensorflow/stream_executor:kernel",
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | BUILD | 20 "//tensorflow/stream_executor/tpu:__subpackages__",
195 "//tensorflow/stream_executor/tpu:tpu_executor",
196 "//tensorflow/stream_executor/tpu:tpu_executor_c_api_hdrs",
211 "//tensorflow/stream_executor/tpu:tpu_executor_c_api_hdrs",
221 "//tensorflow/stream_executor/tpu:tpu_executor_c_api_hdrs",
332 "//tensorflow/stream_executor:device_memory",
333 "//tensorflow/stream_executor:stream",
334 "//tensorflow/stream_executor/lib",
335 "//tensorflow/stream_executor/tpu:c_api_conversions",
336 "//tensorflow/stream_executor/tpu:status_helper",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | jitrt_custom_calls.h | 78 XLA_RUNTIME_REGISTER_ENUM_ATTR_DECODING(stream_executor::dnn::ActivationMode);
79 XLA_RUNTIME_REGISTER_ENUM_ATTR_DECODING(stream_executor::fft::Type);
81 stream_executor::cuda::BlasLt::Epilogue);
140 ::stream_executor::KernelBase* Get(
141 ::stream_executor::StreamExecutor* executor, const char* data,
144 ::stream_executor::KernelBase* Set(
145 ::stream_executor::StreamExecutor* executor, const char* data,
147 std::unique_ptr<::stream_executor::KernelBase> kernel);
152 using Key = std::tuple<::stream_executor::StreamExecutor*, const char*,
154 llvm::SmallDenseMap<Key, std::unique_ptr<::stream_executor::KernelBase>>
[all …]
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 7 import "tensorflow/compiler/xla/stream_executor/dnn.proto";
14 stream_executor.dnn.ConvolutionKind kind = 1;
15 stream_executor.dnn.TensorDescriptorProto input = 2;
16 stream_executor.dnn.TensorDescriptorProto filter = 3;
17 stream_executor.dnn.TensorDescriptorProto output = 4;
18 stream_executor.dnn.ConvolutionDescriptorProto conv_desc = 5;
26 stream_executor.dnn.ActivationMode activation = 8;
|