| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | BUILD | 5 "//tensorflow/stream_executor:build_defs.bzl",
39 "//tensorflow/stream_executor/gpu:gpu_diagnostics_header",
40 "//tensorflow/stream_executor/lib",
41 "//tensorflow/stream_executor/platform",
54 "//tensorflow/stream_executor:device_options",
55 "//tensorflow/stream_executor/gpu:gpu_driver_header",
56 "//tensorflow/stream_executor/lib",
57 "//tensorflow/stream_executor/platform",
58 "//tensorflow/stream_executor/platform:dso_loader",
70 "//tensorflow/stream_executor",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 6 "//tensorflow/stream_executor:build_defs.bzl",
45 deps = ["//tensorflow/stream_executor:platform"],
58 "//tensorflow/stream_executor", # buildcleaner: keep
59 "//tensorflow/stream_executor:executor_cache",
60 "//tensorflow/stream_executor:multi_platform_manager",
61 "//tensorflow/stream_executor:stream_executor_pimpl_header",
62 "//tensorflow/stream_executor/lib",
63 "//tensorflow/stream_executor/platform",
83 "//tensorflow/stream_executor/gpu:gpu_diagnostics_header",
84 "//tensorflow/stream_executor/lib",
[all …]
|
| D | cupti_stub.cc | 27 auto handle_or = stream_executor::internal::DsoLoader::GetCuptiDsoHandle(); in GetDsoHandle()
38 stream_executor::port::Env::Default() in LoadSymbol()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 5 "//tensorflow/stream_executor:build_defs.bzl",
18 "//tensorflow/stream_executor:__subpackages__",
35 deps = ["//tensorflow/stream_executor/platform"],
45 "//tensorflow/stream_executor",
46 "//tensorflow/stream_executor:stream_executor_internal",
47 "//tensorflow/stream_executor/platform",
55 "//tensorflow/stream_executor/lib",
56 "//tensorflow/stream_executor/platform",
65 "//tensorflow/stream_executor:device_options",
66 "//tensorflow/stream_executor/lib",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | BUILD | 4 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
35 "//tensorflow/stream_executor:platform",
51 "//tensorflow/stream_executor:executor_cache",
52 "//tensorflow/stream_executor:multi_platform_manager",
53 "//tensorflow/stream_executor:stream_executor_headers",
54 "//tensorflow/stream_executor/lib",
55 "//tensorflow/stream_executor/platform",
72 "//tensorflow/stream_executor:kernel",
73 "//tensorflow/stream_executor/lib",
87 "//tensorflow/stream_executor:stream_executor_internal",
[all …]
|
| /external/tensorflow/tensorflow/core/platform/default/build_config/ |
| D | BUILD | 32 name = "stream_executor",
34 "//tensorflow/stream_executor/cuda:cuda_activation",
35 "//tensorflow/stream_executor/rocm:rocm_activation",
38 "//tensorflow/stream_executor",
39 "//tensorflow/stream_executor:dnn",
40 "//tensorflow/stream_executor:event",
41 "//tensorflow/stream_executor:multi_platform_manager",
42 "//tensorflow/stream_executor:scratch_allocator",
43 "//tensorflow/stream_executor/cuda:cuda_platform_id",
44 "//tensorflow/stream_executor/host:host_platform_id",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | BUILD | 10 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
38 "//tensorflow/stream_executor/platform",
49 "//tensorflow/stream_executor/lib",
50 "//tensorflow/stream_executor/platform",
91 "//tensorflow/stream_executor/lib",
92 "//tensorflow/stream_executor/platform",
126 "stream_executor.h",
139 "//tensorflow/stream_executor/lib",
140 "//tensorflow/stream_executor/platform",
154 "//tensorflow/stream_executor/platform",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.h | 27 namespace stream_executor {
72 stream_executor::DeviceMemoryAllocator* allocator);
73 stream_executor::DeviceMemoryAllocator* allocator() const;
86 ExecutableRunOptions& set_stream(stream_executor::Stream* stream);
87 stream_executor::Stream* stream() const;
93 stream_executor::Stream* stream);
94 stream_executor::Stream* host_to_device_stream() const;
123 stream_executor::DeviceMemoryAllocator* allocator_ = nullptr;
126 stream_executor::Stream* stream_ = nullptr;
130 stream_executor::Stream* host_to_device_stream_ = nullptr;
|
| D | executable_run_options.cc | 42 stream_executor::DeviceMemoryAllocator* allocator) { in set_allocator()
47 stream_executor::DeviceMemoryAllocator* ExecutableRunOptions::allocator() in allocator()
53 stream_executor::Stream* stream) { in set_stream()
58 stream_executor::Stream* ExecutableRunOptions::stream() const { in stream()
63 stream_executor::Stream* stream) { in set_host_to_device_stream()
68 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream()
|
| /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/mlir/tensorflow/translate/ |
| D | import_model.h | 36 stream_executor::port::StatusOr<mlir::OwningModuleRef> ConvertGraphdefToMlir(
43 stream_executor::port::StatusOr<mlir::OwningModuleRef> ConvertGraphToMlir(
50 stream_executor::port::StatusOr<mlir::OwningModuleRef> ConvertSavedModelToMlir(
56 stream_executor::port::StatusOr<mlir::OwningModuleRef>
|
| D | export_graphdef.h | 32 using stream_executor::port::StatusOr;
42 stream_executor::port::Status ConvertMlirToGraph(
50 stream_executor::port::Status ConvertMlirToGraph(
|
| /external/tensorflow/tensorflow/stream_executor/platform/default/ |
| D | BUILD | 3 package(default_visibility = ["//tensorflow/stream_executor:__subpackages__"])
24 "//tensorflow/stream_executor:platform",
25 "//tensorflow/stream_executor/lib",
26 "//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/gpu/ |
| D | hlo_algorithm_blacklist_test.cc | 52 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(0, false), list[0]); in TEST_F()
53 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(0, true), list[1]); in TEST_F()
54 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(1, false), list[2]); in TEST_F()
55 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(1, true), list[3]); in TEST_F()
|
| D | BUILD | 362 "//tensorflow/stream_executor:device_memory_allocator",
429 "//tensorflow/stream_executor/cuda:cuda_activation",
430 "//tensorflow/stream_executor/cuda:cuda_gpu_executor",
561 "//tensorflow/stream_executor",
562 "//tensorflow/stream_executor:blas",
563 "//tensorflow/stream_executor:device_memory",
564 "//tensorflow/stream_executor:device_memory_allocator",
565 "//tensorflow/stream_executor:kernel",
566 "//tensorflow/stream_executor/gpu:gpu_stream",
577 "//tensorflow/stream_executor/cuda:cuda_stream",
[all …]
|
| D | gpu_executable.h | 116 stream_executor::Stream* stream);
161 std::map<stream_executor::StreamExecutor*, se::ScopedModuleHandle>
163 std::map<stream_executor::StreamExecutor*, BufferAllocToDeviceMemoryMap>
|
| /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 | array_slice.h | 21 namespace stream_executor {
38 using namespace stream_executor; // NOLINT[build/namespaces] variable
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | BUILD | 86 "//tensorflow/core:stream_executor",
125 "//tensorflow/stream_executor:device_memory",
126 "//tensorflow/stream_executor:device_memory_allocator",
158 "//tensorflow/core:stream_executor",
191 "//tensorflow/stream_executor:tf_allocator_adapter",
269 "//tensorflow/stream_executor:device_memory",
270 "//tensorflow/stream_executor:platform",
271 "//tensorflow/stream_executor/cuda:cuda_platform_id",
272 "//tensorflow/stream_executor/host:host_platform_id",
349 "//tensorflow/stream_executor:device_memory_allocator",
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/platform/ |
| D | BUILD | 2 load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends")
26 ] + tf_platform_deps("platform", "//tensorflow/stream_executor/platform/"),
34 ] + tf_platform_deps("dso_loader", "//tensorflow/stream_executor/platform/"),
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 40 namespace stream_executor {
73 virtual stream_executor::Stream* stream() const { return nullptr; } in stream()
75 stream_executor::Stream* stream) const { in MaintainLifetimeOnStream()
107 virtual Status ThenExecute(Device* device, stream_executor::Stream* stream, in ThenExecute()
154 stream_executor::Stream* stream = nullptr;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_init.h | 22 namespace stream_executor {
36 stream_executor::Platform* GPUMachineManager();
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | prepare_reference_module.h | 37 const ::stream_executor::Platform::Id& test_platform_id,
40 const ::stream_executor::Platform::Id&,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 109 TF_ASSIGN_OR_RETURN(auto executor, stream_executor(device_ordinal)); in BorrowStream()
166 StatusOr<se::StreamExecutor*> Backend::stream_executor( in stream_executor() function in xla::Backend
190 stream_executor(device_ordinal_a)); in devices_equivalent()
192 stream_executor(device_ordinal_b)); in devices_equivalent()
|