| /external/executorch/backends/vulkan/tools/gpuinfo/include/ |
| D | app.h | 29 uint32_t subgroup_size; variable
|
| /external/mesa3d/src/vulkan/runtime/ |
| D | vk_nir.c | 121 enum gl_subgroup_size subgroup_size, in vk_spirv_to_nir()
|
| D | vk_acceleration_structure.h | 97 uint32_t subgroup_size; member
|
| D | vk_shader.c | 119 enum gl_subgroup_size subgroup_size = vk_get_subgroup_size( in vk_shader_to_nir() local
|
| /external/OpenCL-CTS/test_conformance/subgroups/ |
| D | subhelpers.cpp | 200 int subgroup_size, int &workgroup_size, in set_last_workgroup_params()
|
| D | subhelpers.h | 65 size_t subgroup_size; member
|
| /external/mesa3d/src/asahi/compiler/ |
| D | agx_nir_lower_subgroups.c | 111 nir_def *subgroup_size = nir_imm_int(b, 32); in lower() local
|
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_screen.c | 253 uint32_t *subgroup_size = ret; in llvmpipe_get_compute_param() local
259 uint32_t *subgroup_size = ret; in llvmpipe_get_compute_param() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | mean.cc | 101 const uint32_t subgroup_size = ctx.gpu_info->vulkan_info.subgroup_size; in GenerateSubgroupBasedMean() local
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_get.c | 949 unsigned subgroup_size; in si_get_compute_param() local
962 uint32_t *subgroup_size = ret; in si_get_compute_param() local
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_subgroups.c | 82 nir_def *subgroup_size = nir_load_subgroup_size(b); in build_subgroup_mask() local
505 unsigned subgroup_size = get_max_subgroup_size(options); in lower_boolean_shuffle() local
857 unsigned subgroup_size = get_max_subgroup_size(options); in lower_scan_reduce() local
|
| /external/mesa3d/src/gallium/frontends/rusticl/mesa/compiler/ |
| D | nir.rs | 322 pub fn subgroup_size(&self) -> u8 { in subgroup_size() method
323 let subgroup_size = unsafe { (*self.nir.as_ptr()).info.subgroup_size }; in subgroup_size() localVariable
|
| /external/mesa3d/src/compiler/spirv/ |
| D | nir_spirv.h | 72 enum gl_subgroup_size subgroup_size; member
|
| /external/mesa3d/src/gallium/frontends/clover/core/ |
| D | device.cpp | 423 device::subgroup_size() const { in subgroup_size() function in device
|
| /external/mesa3d/src/intel/compiler/ |
| D | brw_nir_lower_cooperative_matrix.c | 60 unsigned subgroup_size; member
796 brw_nir_lower_cmat(nir_shader *shader, unsigned subgroup_size) in brw_nir_lower_cmat()
|
| /external/mesa3d/src/gallium/frontends/rusticl/core/ |
| D | kernel.rs | 341 subgroup_size: usize, field
1070 let subgroup_size = nir.subgroup_size(); in new() localVariable
1706 pub fn subgroup_size(&self) -> usize { in subgroup_size() method
1762 let subgroup_size = self.subgroup_size_for_block(dev, block); in subgroups_for_block() localVariable
|
| /external/mesa3d/src/intel/vulkan/grl/gpu/ |
| D | morton_radix_sort.h | 365 const uint subgroup_size = get_sub_group_size(); in sort_morton_codes_scatter_items() local
458 const uint subgroup_size = get_sub_group_size(); in sort_morton_codes_scatter_items() local
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_shader_object.c | 97 const VkShaderRequiredSubgroupSizeCreateInfoEXT *const subgroup_size = in radv_shader_stage_init() local
|
| D | radv_pipeline.c | 172 const VkPipelineShaderStageRequiredSubgroupSizeCreateInfo *const subgroup_size = in radv_pipeline_get_shader_key() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | gpu_info.cc | 823 for (auto subgroup_size : supported_subgroup_sizes) { in SupportsSubGroupWithSize() local
|
| /external/mesa3d/src/freedreno/ir3/ |
| D | ir3_shader.c | 1037 gl_shader_stage stage, unsigned *subgroup_size, in ir3_shader_get_subgroup_size()
|
| D | ir3_lower_subgroups.c | 646 unsigned subgroup_size, max_subgroup_size; in lower_scan_reduce() local
|
| D | ir3_nir.c | 818 unsigned subgroup_size = 0, max_subgroup_size = 0; in ir3_nir_post_finalize() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util.cc | 1677 int64_t subgroup_size = sharding.subgroup_types().size(); in GroupShardingOnDims() local
1705 int64_t subgroup_size = sharding.subgroup_types().size(); in GetManualSubgroupSharding() local
|
| D | hlo_verifier.cc | 369 int64_t subgroup_size = GetSubgroupSize(ag, group_mode); in CheckCommonAllGatherInvariants() local
448 int64_t subgroup_size = GetSubgroupSize(ars, group_mode); in HandleReduceScatter() local
|