Searched refs:opencl_info (Results 1 – 6 of 6) sorted by relevance
140 info.opencl_info.cl_version = ParseCLVersion(opencl_c_version); in GpuInfoFromDeviceID()141 info.opencl_info.extensions = in GpuInfoFromDeviceID()143 info.opencl_info.supports_fp16 = false; in GpuInfoFromDeviceID()144 info.opencl_info.supports_image3d_writes = false; in GpuInfoFromDeviceID()145 for (const auto& ext : info.opencl_info.extensions) { in GpuInfoFromDeviceID()147 info.opencl_info.supports_fp16 = true; in GpuInfoFromDeviceID()150 info.opencl_info.supports_image3d_writes = true; in GpuInfoFromDeviceID()154 info.opencl_info.supports_images = in GpuInfoFromDeviceID()159 info.opencl_info.supports_fp32_rtn = f32_config & CL_FP_ROUND_TO_NEAREST; in GpuInfoFromDeviceID()161 if (info.opencl_info.supports_fp16) { in GpuInfoFromDeviceID()[all …]
57 info->opencl_info.supports_r_f16_tex2d = in AddSupportedImageFormats()58 info->opencl_info.supports_r_f16_tex2d || in AddSupportedImageFormats()60 info->opencl_info.supports_rg_f16_tex2d = in AddSupportedImageFormats()61 info->opencl_info.supports_rg_f16_tex2d || in AddSupportedImageFormats()63 info->opencl_info.supports_rgb_f16_tex2d = in AddSupportedImageFormats()64 info->opencl_info.supports_rgb_f16_tex2d || in AddSupportedImageFormats()66 info->opencl_info.supports_rgba_f16_tex2d = in AddSupportedImageFormats()67 info->opencl_info.supports_rgba_f16_tex2d || in AddSupportedImageFormats()69 info->opencl_info.supports_r_f32_tex2d = in AddSupportedImageFormats()70 info->opencl_info.supports_r_f32_tex2d || in AddSupportedImageFormats()[all …]
401 return opencl_info.supports_fp16_rtn || opencl_info.supports_fp32_rtn; in IsRoundToNearestSupported()411 return opencl_info.supports_fp16; in SupportsFP16()421 return opencl_info.cl_version >= OpenClVersion::kCl1_2; in SupportsTextureArray()431 return opencl_info.cl_version >= OpenClVersion::kCl1_2; in SupportsImageBuffer()445 return opencl_info.supports_image3d_writes; in SupportsImage3D()452 return opencl_info.supports_images; in SupportsImages()469 extensions = &opencl_info.extensions; in SupportsExtension()494 return data_type == DataType::FLOAT32 ? opencl_info.supports_r_f32_tex2d in SupportsFloatImage2D()495 : opencl_info.supports_r_f16_tex2d; in SupportsFloatImage2D()497 return data_type == DataType::FLOAT32 ? opencl_info.supports_rg_f32_tex2d in SupportsFloatImage2D()[all …]
385 OpenClInfo opencl_info; member
41 if (gpu_info.opencl_info.cl_version < OpenClVersion::kCl1_2 && in CanCreateTensorWithShape()
467 if (gpu_info.opencl_info.cl_version == OpenClVersion::kCl2_0) { in GenerateConv()