| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.cc | 27 /* static */ const char *DeviceDescription::kUndefinedString = "<undefined>";
29 DeviceDescription::DeviceDescription() in DeviceDescription() function in stream_executor::DeviceDescription
59 std::unique_ptr<std::map<std::string, std::string>> DeviceDescription::ToMap() in ToMap()
110 : device_description_(new DeviceDescription) {} in DeviceDescriptionBuilder()
114 bool DeviceDescription::cuda_compute_capability(int *major, int *minor) const { in cuda_compute_capability()
120 bool DeviceDescription::rocm_amdgpu_isa_version(int *version) const { in rocm_amdgpu_isa_version()
129 bool ThreadDimOk(const DeviceDescription &device_description, in ThreadDimOk()
154 void CalculateDimensionality(const DeviceDescription &device_description, in CalculateDimensionality()
|
| D | device_description.h | 40 class DeviceDescription {
173 DeviceDescription();
220 SE_DISALLOW_COPY_AND_ASSIGN(DeviceDescription);
322 std::unique_ptr<DeviceDescription> Build() { in Build()
327 std::unique_ptr<DeviceDescription> device_description_;
337 bool ThreadDimOk(const DeviceDescription &device_description,
348 void CalculateDimensionality(const DeviceDescription &device_description,
|
| D | platform.h | 36 class DeviceDescription; variable
144 virtual port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| D | stream_executor_pimpl.h | 327 const DeviceDescription &GetDeviceDescription() const;
637 std::unique_ptr<DeviceDescription> CreateDeviceDescription() const;
700 mutable std::unique_ptr<DeviceDescription> device_description_
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 77 int CalculateOccupancy(const DeviceDescription& device_description,
84 const DeviceDescription& device_description,
198 port::StatusOr<std::unique_ptr<DeviceDescription>> CreateDeviceDescription() in CreateDeviceDescription()
203 static port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_atomic_test.cc | 112 const se::DeviceDescription& device_description = in TEST_F()
|
| D | reduction_vectorization_test.cc | 339 const se::DeviceDescription& device_description = in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 135 port::StatusOr<std::unique_ptr<DeviceDescription>> CreateDeviceDescription() in CreateDeviceDescription()
140 static port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| D | host_platform.h | 54 port::StatusOr<std::unique_ptr<DeviceDescription>> DescriptionForDevice(
|
| D | host_platform.cc | 44 port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.h | 43 port::StatusOr<std::unique_ptr<DeviceDescription>> DescriptionForDevice(
|
| D | executor.h | 168 port::StatusOr<std::unique_ptr<DeviceDescription>> CreateDeviceDescription() in CreateDeviceDescription()
173 static port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| D | platform.cc | 45 port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| D | executor.cc | 116 port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.h | 67 port::StatusOr<std::unique_ptr<DeviceDescription>> DescriptionForDevice(
|
| D | cuda_platform.cc | 144 port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| D | cuda_gpu_executor.cc | 457 const DeviceDescription& device_description = in VlogOccupancyInfo()
481 int GpuExecutor::CalculateOccupancy(const DeviceDescription& device_description, in CalculateOccupancy()
498 const DeviceDescription& device_description, in CompareOccupancy()
961 port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.h | 67 port::StatusOr<std::unique_ptr<DeviceDescription>> DescriptionForDevice(
|
| D | rocm_platform.cc | 107 port::StatusOr<std::unique_ptr<DeviceDescription>>
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 82 // stream_executor::DeviceDescription::pci_bus_id.
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 309 se::port::StatusOr<std::unique_ptr<se::DeviceDescription>> in TEST()
311 return std::unique_ptr<se::DeviceDescription>(nullptr); in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_platform.h | 80 StatusOr<std::unique_ptr<::stream_executor::DeviceDescription>>
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 234 const se::DeviceDescription& description = in BuildLocalDevices()
254 std::unique_ptr<xla::se::DeviceDescription> desc, in BuildDistributedDevices()
|
| /external/autotest/client/common_lib/cros/network/ |
| D | interface.py | 24 DeviceDescription = collections.namedtuple('DeviceDescription', variable
299 return DeviceDescription(device_name, module_path)
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_internal.h | 63 port::StatusOr<std::unique_ptr<DeviceDescription>> DescriptionForDevice(
|