Home
last modified time | relevance | path

Searched refs:device_description (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc129 bool ThreadDimOk(const DeviceDescription &device_description, in ThreadDimOk() argument
133 device_description.threads_per_block_limit(); in ThreadDimOk()
140 const auto &limit = device_description.thread_dim_limit(); in ThreadDimOk()
154 void CalculateDimensionality(const DeviceDescription &device_description, in CalculateDimensionality() argument
157 *threads_per_block = device_description.threads_per_block_limit(); in CalculateDimensionality()
DBUILD46 name = "device_description",
47 srcs = ["device_description.cc"],
48 hdrs = ["device_description.h"],
62 "device_description.h",
116 "device_description.h",
188 "device_description.h",
318 "device_description.h",
463 ":device_description",
483 "device_description.h",
551 "device_description.h",
[all …]
Ddevice_description.h337 bool ThreadDimOk(const DeviceDescription &device_description,
348 void CalculateDimensionality(const DeviceDescription &device_description,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_atomic_test.cc112 const se::DeviceDescription& device_description = in TEST_F() local
115 device_description.cuda_compute_capability(&cc_major, &cc_minor); in TEST_F()
Dreduction_vectorization_test.cc339 const se::DeviceDescription& device_description = in TEST_F() local
342 device_description.cuda_compute_capability(&cc_major, &cc_minor); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.cc267 const se::DeviceDescription* device_description = in ExecuteWrapperAfterExecution() local
270 stream->ThenDoHostCallback([profile, device_description]() { in ExecuteWrapperAfterExecution()
272 profile->ToString(device_description->clock_rate_ghz())); in ExecuteWrapperAfterExecution()
DBUILD1155 "//tensorflow/stream_executor:device_description",
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler.py294 device_description = (
298 pprof_proto.string_table.append(device_description)
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h77 int CalculateOccupancy(const DeviceDescription& device_description,
84 const DeviceDescription& device_description,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc457 const DeviceDescription& device_description = in VlogOccupancyInfo() local
463 int blocks_per_sm = CalculateOccupancy(device_description, regs_per_thread, in VlogOccupancyInfo()
468 CompareOccupancy(&blocks_per_sm, device_description, regs_per_thread, in VlogOccupancyInfo()
481 int GpuExecutor::CalculateOccupancy(const DeviceDescription& device_description, in CalculateOccupancy() argument
498 const DeviceDescription& device_description, in CompareOccupancy() argument
/external/webrtc/modules/desktop_capture/mac/
Ddesktop_configuration.mm53 NSDictionary* device_description = [screen deviceDescription];
55 [[device_description objectForKey:@"NSScreenNumber"] intValue]);
/external/autotest/client/common_lib/cros/network/
Dinterface.py277 def device_description(self): member in Interface
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc348 int GpuExecutor::CalculateOccupancy(const DeviceDescription& device_description, in CalculateOccupancy() argument
358 const DeviceDescription& device_description, in CompareOccupancy() argument
/external/autotest/server/cros/network/
Dwifi_client.py369 self._interface.device_description)
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD694 "//tensorflow/stream_executor:device_description",