Home
last modified time | relevance | path

Searched refs:GpuDeviceInfo (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h42 std::string platform_name, GpuDeviceInfo gpu_device_info, in IrEmitterContext()
64 GpuDeviceInfo gpu_device_info() const { return gpu_device_info_; } in gpu_device_info()
92 GpuDeviceInfo gpu_device_info_;
Dlaunch_dimensions.cc37 static int64 ThreadsPerBlockLimit(GpuDeviceInfo gpu_device_info) { in ThreadsPerBlockLimit()
58 GpuDeviceInfo gpu_device_info, in CalculateLaunchDimensions()
Dgpu_device_info.h31 struct GpuDeviceInfo { struct
Dgpu_compiler.h149 GpuDeviceInfo GetGpuDeviceInfo(se::StreamExecutor* stream_exec);
156 const std::string& platform_name, GpuDeviceInfo gpu_device_info,
Dlaunch_dimensions.h69 GpuDeviceInfo gpu_device_info,
Dgpu_compiler.cc551 const std::string& platform_name, GpuDeviceInfo gpu_device_info, in CompileModuleToLlvmIrImpl()
867 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in RunBackend()
963 GpuDeviceInfo GetGpuDeviceInfo(se::StreamExecutor* stream_exec) { in GetGpuDeviceInfo()
964 GpuDeviceInfo gpu_device_info; in GetGpuDeviceInfo()
986 const std::string& platform_name, GpuDeviceInfo gpu_device_info, in CompileModuleToLlvmIr()
Dbuffer_comparator.cc608 GpuDeviceInfo gpu_device_info; in DeviceCompare()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h153 struct GpuDeviceInfo { struct
162 void set_tensorflow_gpu_device_info(GpuDeviceInfo* g) { in set_tensorflow_gpu_device_info() argument
166 virtual const GpuDeviceInfo* tensorflow_gpu_device_info() const { in tensorflow_gpu_device_info()
279 GpuDeviceInfo* gpu_device_info_ = nullptr;
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc63 const DeviceBase::GpuDeviceInfo** dev_info, in PrepareCopy()
118 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in SetProtoFromGPU()
192 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in DeviceToDeviceCopy()
259 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToCPU()
304 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyCPUTensorToGPU()
428 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToSameGPU()
Dgpu_util_platform_specific.cc52 const DeviceBase::GpuDeviceInfo* gpu_info = in ThenExecute()
Dgpu_device.h147 GpuDeviceInfo* gpu_device_info_ = nullptr;
Dgpu_device.cc472 gpu_device_info_ = new GpuDeviceInfo; in Init()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc139 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
145 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
Drenamed_device.h53 const GpuDeviceInfo* tensorflow_gpu_device_info() const override { in tensorflow_gpu_device_info()
Dring_gatherer.cc140 const DeviceBase::GpuDeviceInfo* gpu_info = in RunAsyncParts()
Dring_reducer.cc190 const DeviceBase::GpuDeviceInfo* gpu_info = in RunAsyncParts()
Dring_alg.cc255 const DeviceBase::GpuDeviceInfo* gpu_device_info = in TensorDebugString()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc46 xla::gpu::GpuDeviceInfo gpu_device_info{}; in CompileAndPrintLlvmIr()
Dmlir_gpu_test_base.cc54 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in CompileMlirModule()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h69 const DeviceBase::GpuDeviceInfo* info, const Tensor* input, in Participant()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h258 std::unique_ptr<GpuDeviceInfo> gpu_device_info_ TF_GUARDED_BY(mu_);
Dxla_device.cc365 auto gpu_device_info = absl::make_unique<GpuDeviceInfo>(); in GetDeviceContextLocked()
/external/tensorflow/tensorflow/core/kernels/
Dfunctional_ops.cc404 const DeviceBase::GpuDeviceInfo* gpu_device_info = in CondResultToBool()