| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 56 min_numa_node_ = exec->GetDeviceDescription().numa_node(); in InspectNumaNodes()
60 std::min(min_numa_node_, exec->GetDeviceDescription().numa_node()); in InspectNumaNodes()
62 exec->GetDeviceDescription().numa_node() + 1); in InspectNumaNodes()
77 return exec->GetDeviceDescription().numa_node() - min_numa_node_; in DeviceToBus()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 91 min_numa_node_ = exec->GetDeviceDescription().numa_node(); in InspectNumaNodes()
95 std::min(min_numa_node_, exec->GetDeviceDescription().numa_node()); in InspectNumaNodes()
97 exec->GetDeviceDescription().numa_node() + 1); in InspectNumaNodes()
112 return exec->GetDeviceDescription().numa_node() - min_numa_node_; in DeviceToBus()
|
| D | cuda_blas.cc | 1599 stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major, in DoBlasGemm()
1896 if (stream->parent()->GetDeviceDescription().cuda_compute_capability( in DoBlasGemmWithAlgorithmImpl()
2214 if (stream->parent()->GetDeviceDescription().cuda_compute_capability( in DoBlasGemmBatchedInternal()
2367 if (stream->parent()->GetDeviceDescription().cuda_compute_capability( in DoBlasGemmStridedBatched()
|
| D | cuda_gpu_executor.cc | 485 kernel.parent()->GetDeviceDescription(); in VlogOccupancyInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 220 return (executor_a->GetDeviceDescription().name() == in devices_equivalent()
221 executor_b->GetDeviceDescription().name()); in devices_equivalent()
|
| D | hlo_execution_profile_test.cc | 69 backend().default_stream_executor()->GetDeviceDescription()), in TEST_F()
|
| D | executable.cc | 140 profile_ptr->ToString(stream->parent()->GetDeviceDescription())); in ExecuteOnStreamWrapper()
|
| D | platform_util.cc | 189 const auto& description = executor->GetDeviceDescription(); in IsDeviceSupported()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 41 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetNumAvailableGPUs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.h | 60 stream->parent()->GetDeviceDescription().name()) == in WillAutotuneKernel()
|
| D | hlo_execution_profiler.cc | 64 clock_rate_ghz_ = stream->parent()->GetDeviceDescription().clock_rate_ghz(); in HloExecutionProfiler()
|
| D | nvptx_compiler.cc | 682 &stream_exec->GetDeviceDescription(), in RunBackend()
738 if (!stream_exec->GetDeviceDescription().cuda_compute_capability(&cc_major, in RunBackend()
780 stream_exec->GetDeviceDescription().memory_bandwidth()); in RunBackend()
|
| D | stream_executor_util.cc | 32 CHECK(stream_executor.GetDeviceDescription().cuda_compute_capability(&major, in IsVoltaOrLater()
|
| D | gpu_executable.cc | 74 executor->GetDeviceDescription().cuda_compute_capability( in ExecuteThunks()
|
| D | cudnn_conv_algorithm_picker.cc | 118 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
|
| D | gemm_thunk.cc | 411 const string& device_name = stream->parent()->GetDeviceDescription().name(); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 124 build_executor->GetDeviceDescription().name(), in ValidateExecutionOptions()
126 run_executor->GetDeviceDescription().name()); in ValidateExecutionOptions()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 293 const DeviceDescription &StreamExecutor::GetDeviceDescription() const { in GetDeviceDescription() function in stream_executor::StreamExecutor
331 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveAlgorithms()
353 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveBackwardDataAlgorithms()
366 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveBackwardFilterAlgorithms()
|
| D | stream_executor_pimpl.h | 347 const DeviceDescription &GetDeviceDescription() const;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 905 if (!se->GetDeviceDescription().cuda_compute_capability(&cc_major, in SingleVirtualDeviceMemoryLimit()
1197 const se::DeviceDescription& desc = se->GetDeviceDescription(); in CreateGPUDevice()
1292 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetDeviceLocalities()
1367 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetDefaultMinGPUMultiprocessorCount()
1542 const auto& description = stream_exec->GetDeviceDescription(); in GetValidDeviceIds()
1613 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetValidDeviceIds()
|
| D | gpu_process_state.cc | 74 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForGPU()
|
| D | gpu_device_test.cc | 49 if (!se->GetDeviceDescription().cuda_compute_capability(cc_major, cc_minor)) { in GetComputeCapability()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 49 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 305 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
461 stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major, in launch()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 185 hlo_execution_profile.ToString(executor->GetDeviceDescription()); in ExecuteAndFetchProfile()
|