| /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/stream_executor/cuda/ |
| D | BUILD | 7 "//tensorflow/stream_executor:build_defs.bzl",
49 deps = ["//tensorflow/stream_executor:platform"],
62 "//tensorflow/stream_executor", # buildcleaner: keep
63 "//tensorflow/stream_executor:executor_cache",
64 "//tensorflow/stream_executor:multi_platform_manager",
65 "//tensorflow/stream_executor:stream_executor_pimpl_header",
66 "//tensorflow/stream_executor/lib",
67 "//tensorflow/stream_executor/platform",
88 "//tensorflow/stream_executor/gpu:gpu_diagnostics_header",
89 "//tensorflow/stream_executor/lib",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 6 "//tensorflow/stream_executor:build_defs.bzl",
23 "//tensorflow/stream_executor:__subpackages__",
40 deps = ["//tensorflow/stream_executor/platform"],
50 "//tensorflow/stream_executor",
51 "//tensorflow/stream_executor:stream_executor_internal",
52 "//tensorflow/stream_executor/platform",
60 "//tensorflow/stream_executor/lib",
61 "//tensorflow/stream_executor/platform",
70 "//tensorflow/stream_executor:device_options",
71 "//tensorflow/stream_executor/lib",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | BUILD | 6 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
37 "//tensorflow/stream_executor:platform",
54 "//tensorflow/stream_executor:executor_cache",
55 "//tensorflow/stream_executor:multi_platform_manager",
56 "//tensorflow/stream_executor:stream_executor_headers",
57 "//tensorflow/stream_executor/lib",
58 "//tensorflow/stream_executor/platform",
76 "//tensorflow/stream_executor:kernel",
77 "//tensorflow/stream_executor/lib",
91 "//tensorflow/stream_executor:stream_executor_internal",
[all …]
|
| /external/tensorflow/tensorflow/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>;
81 StatusOr<::stream_executor::StreamExecutor*> TpuPlatform::GetExecutor( in GetExecutor()
82 const ::stream_executor::StreamExecutorConfig& config) { in GetExecutor()
87 StatusOr<std::unique_ptr<::stream_executor::StreamExecutor>>
89 const ::stream_executor::StreamExecutorConfig& config) { in GetUncachedExecutor()
103 return std::make_unique<stream_executor::StreamExecutor>( in GetUncachedExecutor()
107 ::stream_executor::Platform::Id TpuPlatform::id() const { in id()
136 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 | 26 "//tensorflow/stream_executor:stream_executor_headers",
59 "//tensorflow/stream_executor:device_memory",
60 "//tensorflow/stream_executor:device_memory_allocator",
121 "//tensorflow/stream_executor",
122 "//tensorflow/stream_executor/lib",
123 "//tensorflow/stream_executor/platform",
137 "//tensorflow/stream_executor:stream_header",
161 "//tensorflow/stream_executor:stream_executor_headers",
162 "//tensorflow/stream_executor/lib",
171 deps = ["//tensorflow/stream_executor:stream_header"],
[all …]
|
| D | c_api_conversions.h | 39 SE_DeviceMemoryBase ToC(const stream_executor::DeviceMemoryBase& base);
40 void ToC(const stream_executor::DeviceMemoryBase& base,
42 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);
76 SE_DeviceMemoryBase ToC(const stream_executor::DeviceMemoryBase& base);
77 stream_executor::DeviceMemoryBase FromC(const SE_DeviceMemoryBase& se_base);
92 stream_executor::DeviceMemoryAllocator* allocator);
95 SE_DeviceMemoryAllocator ToC(stream_executor::DeviceMemoryAllocator* allocator);
98 SE_MaybeOwningDeviceMemory ToC(stream_executor::OwningDeviceMemory* mem);
|
| D | c_api_conversions.cc | 29 xla::ShapeTree<stream_executor::DeviceMemoryBase> xla_shape_tree( in FromC()
49 const stream_executor::OwningDeviceMemory* owned = in ToC()
59 ToC(static_cast<stream_executor::DeviceMemoryAllocator*>(nullptr)); in ToC()
67 stream_executor::DeviceMemoryAllocator* allocator) { in FromC()
70 stream_executor::OwningDeviceMemory(ApiConverter::FromC(se_mem->memory), in FromC()
78 stream_executor::DeviceMemoryAllocator* allocator) { in ToC()
95 reinterpret_cast<stream_executor::DeviceMemoryAllocator*>(ctx) in ToC()
111 auto status = reinterpret_cast<stream_executor::DeviceMemoryAllocator*>(ctx) in ToC()
122 SE_MaybeOwningDeviceMemory ToC(stream_executor::OwningDeviceMemory* mem) { in ToC()
131 void ToC(const stream_executor::DeviceMemoryBase& base, in ToC()
[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"],
48 "//tensorflow/stream_executor:executor_cache",
49 "//tensorflow/stream_executor:multi_platform_manager",
50 "//tensorflow/stream_executor:platform",
51 "//tensorflow/stream_executor:stream_executor_internal",
52 "//tensorflow/stream_executor:stream_executor_pimpl",
[all …]
|
| D | stream_executor_internal.h | 27 namespace stream_executor {
54 SP_DeviceFns device_fns, SP_StreamExecutor stream_executor,
91 stream_executor::ExecutorCache executor_cache_;
96 CStream(SP_Device* device, SP_StreamExecutor* stream_executor) in CStream() argument
98 stream_executor_(stream_executor), in CStream()
126 CEvent(SP_Device* device, SP_StreamExecutor* stream_executor) in CEvent() argument
128 stream_executor_(stream_executor), in CEvent()
162 CTimer(SP_Device* device, SP_StreamExecutor* stream_executor, in CTimer() argument
165 stream_executor_(stream_executor), in CTimer()
|
| /external/tensorflow/tensorflow/core/platform/default/build_config/ |
| D | BUILD | 31 name = "stream_executor",
33 "//tensorflow/stream_executor/cuda:cuda_activation",
34 "//tensorflow/stream_executor/rocm:rocm_activation",
37 "//tensorflow/stream_executor",
38 "//tensorflow/stream_executor:dnn",
39 "//tensorflow/stream_executor:event",
40 "//tensorflow/stream_executor:multi_platform_manager",
41 "//tensorflow/stream_executor:scratch_allocator",
42 "//tensorflow/stream_executor/cuda:cuda_platform_id",
43 "//tensorflow/stream_executor/host:host_platform_id",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | BUILD | 12 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
40 "//tensorflow/stream_executor/platform",
51 "//tensorflow/stream_executor/lib",
52 "//tensorflow/stream_executor/platform",
91 "//tensorflow/stream_executor/lib",
92 "//tensorflow/stream_executor/platform",
125 "stream_executor.h",
138 "//tensorflow/stream_executor/lib",
139 "//tensorflow/stream_executor/platform",
154 "//tensorflow/stream_executor/platform",
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | BUILD | 16 "//tensorflow/stream_executor/tpu:__subpackages__",
125 "//tensorflow/stream_executor/tpu:tpu_executor_c_api_hdrs",
135 "//tensorflow/stream_executor/tpu:tpu_executor_c_api_hdrs",
156 "//tensorflow/stream_executor/tpu:tpu_executor",
157 "//tensorflow/stream_executor/tpu:tpu_executor_c_api_hdrs",
180 "//tensorflow/stream_executor/tpu:tpu_executor",
181 "//tensorflow/stream_executor/tpu:tpu_executor_c_api_hdrs",
243 "//tensorflow/stream_executor:device_memory",
244 "//tensorflow/stream_executor:stream",
245 "//tensorflow/stream_executor/lib",
[all …]
|
| D | tpu_on_demand_compiler.cc | 61 stream_executor::tpu::SerializeProto(dev_assign_proto); in ToC()
75 const_cast<stream_executor::Stream*>(options.stream())->implementation(); in ToC()
139 stream_executor::tpu::SerializedProto_Free( in ExecuteAsyncOnStream()
191 absl::Span<const stream_executor::DeviceMemoryBase> arguments, in LoadProgramAndEnqueueToStream()
192 stream_executor::DeviceMemoryBase result, in LoadProgramAndEnqueueToStream()
193 absl::optional<stream_executor::DeviceMemoryBase> in LoadProgramAndEnqueueToStream()
214 stream_executor::Platform::Id PlatformId() const override { in PlatformId()
220 stream_executor::StreamExecutor* executor, in RunHloPasses()
224 stream_executor::tpu::SerializedProto_Free(hlo_module.proto); in RunHloPasses()
228 hlo_module.proto = stream_executor::tpu::SerializeProto(module->ToProto()); in RunHloPasses()
[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, PlatformGpuId gpu_id,
77 stream_executor::gpu::GpuContext& gpu_context, PlatformGpuId 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 …]
|
| D | executable_run_options.cc | 44 stream_executor::DeviceMemoryAllocator* allocator) { in set_allocator()
49 stream_executor::DeviceMemoryAllocator* ExecutableRunOptions::allocator() in allocator()
55 stream_executor::Stream* stream) { in set_stream()
60 stream_executor::Stream* ExecutableRunOptions::stream() const { in stream()
65 stream_executor::Stream* stream) { in set_host_to_device_stream()
70 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.h | 39 stream_executor::port::StatusOr<mlir::OwningModuleRef> ConvertGraphdefToMlir(
46 stream_executor::port::StatusOr<mlir::OwningModuleRef> ConvertGraphToMlir(
54 stream_executor::port::StatusOr<mlir::OwningModuleRef> ConvertFunctionToMlir(
60 stream_executor::port::StatusOr<mlir::OwningModuleRef> ConvertSavedModelToMlir(
66 stream_executor::port::StatusOr<mlir::OwningModuleRef>
80 stream_executor::port::StatusOr<mlir::OwningModuleRef>
113 virtual stream_executor::port::StatusOr<const Graph*> GetSubGraph(
129 stream_executor::port::StatusOr<mlir::OwningModuleRef>
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 7 import "tensorflow/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;
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | BUILD | 40 "//tensorflow/core/platform:stream_executor",
86 "//tensorflow/stream_executor:device_memory",
87 "//tensorflow/stream_executor:device_memory_allocator",
88 "//tensorflow/stream_executor:event",
106 "//tensorflow/stream_executor:device_memory",
107 "//tensorflow/stream_executor:device_memory_allocator",
123 "//tensorflow/core/platform:stream_executor",
125 "//tensorflow/stream_executor:event",
209 "//tensorflow/stream_executor:event",
210 "//tensorflow/stream_executor:stream",
[all …]
|