Searched refs:apple_info (Results 1 – 3 of 3) sorted by relevance
667 int GetMaximumPossibleWavesCount(const AppleInfo& apple_info, in GetMaximumPossibleWavesCount() argument669 if (apple_info.IsLocalMemoryPreferredOverGlobal()) { in GetMaximumPossibleWavesCount()676 int GetRecommendedBlockSize(const AppleInfo& apple_info, in GetRecommendedBlockSize() argument678 const int max_waves = GetMaximumPossibleWavesCount(apple_info, dst_shape); in GetRecommendedBlockSize()679 const int cu_count = apple_info.GetComputeUnitsCount(); in GetRecommendedBlockSize()692 const AppleInfo& apple_info, const Convolution2DAttributes& attr, in GetConvParamsForA7A8() argument709 int blk_total_size = GetRecommendedBlockSize(apple_info, dst_shape); in GetConvParamsForA7A8()772 const AppleInfo& apple_info, const Convolution2DAttributes& attr, in GetConvParamsForA9AndHigher() argument776 int blk_total_size = GetRecommendedBlockSize(apple_info, dst_shape); in GetConvParamsForA9AndHigher()778 if (blk_total_size >= 2 && apple_info.IsBionic()) { in GetConvParamsForA9AndHigher()[all …]
357 gpu_info->apple_info = AppleInfo(lowered); in GetGpuInfoFromDeviceDescription()404 return apple_info.IsRoundToNearestSupported(); in IsRoundToNearestSupported()519 return apple_info.GetComputeUnitsCount(); in GetComputeUnitsCount()
370 AppleInfo apple_info; member