Searched defs:gpu_device_info (Results 1 – 8 of 8) sorted by relevance
38 static int64_t ThreadsPerBlockLimit(GpuDeviceInfo gpu_device_info) { in ThreadsPerBlockLimit()58 GpuDeviceInfo gpu_device_info, in ThreadsPerBlockRowVectorized()80 const Shape& shape, GpuDeviceInfo gpu_device_info, in CalculateLaunchDimensions()
41 std::string platform_name, GpuDeviceInfo gpu_device_info, in IrEmitterContext()63 GpuDeviceInfo gpu_device_info() const { return gpu_device_info_; } in gpu_device_info() function
22 GpuDeviceInfo gpu_device_info; in GetGpuDeviceInfo() local
969 GpuDeviceInfo gpu_device_info, in CompileModuleToLlvmIrImpl()1340 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in RunBackend() local1449 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in CompileAheadOfTime() local1523 GpuDeviceInfo gpu_device_info, in CompileModuleToLlvmIr()
657 GpuDeviceInfo gpu_device_info; in DeviceCompare() local
52 xla::gpu::GpuDeviceInfo gpu_device_info; member
57 xla::gpu::GpuDeviceInfo gpu_device_info{}; in CompileAndPrintLlvmIr() local
58 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in CompileMlirModule() local