Home
last modified time | relevance | path

Searched refs:metal_info (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_device.cc31 gpu_info.metal_info.max_work_group_size_x = threadsPerGroup.width; in CreateGpuInfoFromMetalDevice()
32 gpu_info.metal_info.max_work_group_size_y = threadsPerGroup.height; in CreateGpuInfoFromMetalDevice()
33 gpu_info.metal_info.max_work_group_size_z = threadsPerGroup.depth; in CreateGpuInfoFromMetalDevice()
35 gpu_info.metal_info.max_work_group_size_x = 256; in CreateGpuInfoFromMetalDevice()
36 gpu_info.metal_info.max_work_group_size_y = 256; in CreateGpuInfoFromMetalDevice()
37 gpu_info.metal_info.max_work_group_size_z = 64; in CreateGpuInfoFromMetalDevice()
41 gpu_info.metal_info.buffer_max_size = [device maxBufferLength]; in CreateGpuInfoFromMetalDevice()
44 gpu_info.metal_info.buffer_max_size = 256 * 1024 * 1024; in CreateGpuInfoFromMetalDevice()
48 gpu_info.metal_info.language_version = MetalLanguageVersion::kMetal2_3; in CreateGpuInfoFromMetalDevice()
50 gpu_info.metal_info.language_version = MetalLanguageVersion::kMetal2_2; in CreateGpuInfoFromMetalDevice()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_info.cc535 return metal_info.max_work_group_size_x; in GetMaxWorkGroupSizeForX()
551 return metal_info.max_work_group_size_y; in GetMaxWorkGroupSizeForY()
567 return metal_info.max_work_group_size_z; in GetMaxWorkGroupSizeForZ()
583 int max_size = metal_info.max_work_group_size_x; in GetMaxWorkGroupTotalSize()
584 max_size = std::max(max_size, metal_info.max_work_group_size_y); in GetMaxWorkGroupTotalSize()
585 max_size = std::max(max_size, metal_info.max_work_group_size_z); in GetMaxWorkGroupTotalSize()
655 return metal_info.buffer_max_size; in GetMaxBufferSize()
Dgpu_info.h382 MetalInfo metal_info; member