Searched defs:gpu_device_info (Results 1 – 9 of 9) sorted by relevance
37 static int64 ThreadsPerBlockLimit(GpuDeviceInfo gpu_device_info) { in ThreadsPerBlockLimit()58 GpuDeviceInfo gpu_device_info, in CalculateLaunchDimensions()
42 std::string platform_name, GpuDeviceInfo gpu_device_info, in IrEmitterContext()64 GpuDeviceInfo gpu_device_info() const { return gpu_device_info_; } in gpu_device_info() function
551 const std::string& platform_name, GpuDeviceInfo gpu_device_info, in CompileModuleToLlvmIrImpl()867 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in RunBackend() local964 GpuDeviceInfo gpu_device_info; in GetGpuDeviceInfo() local986 const std::string& platform_name, GpuDeviceInfo gpu_device_info, in CompileModuleToLlvmIr()
608 GpuDeviceInfo gpu_device_info; in DeviceCompare() local
46 xla::gpu::GpuDeviceInfo gpu_device_info{}; in CompileAndPrintLlvmIr() local
54 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in CompileMlirModule() local
255 const DeviceBase::GpuDeviceInfo* gpu_device_info = in TensorDebugString() local
365 auto gpu_device_info = absl::make_unique<GpuDeviceInfo>(); in GetDeviceContextLocked() local
404 const DeviceBase::GpuDeviceInfo* gpu_device_info = in CondResultToBool() local