Searched defs:subgroup_size (Results 1 – 15 of 15) sorted by relevance
| /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/frontends/clover/core/ |
| D | device.cpp | 272 device::subgroup_size() const { in subgroup_size() function in device
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_get.c | 818 uint32_t *subgroup_size = ret; in si_get_compute_param() local
|
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_screen.c | 517 uint32_t *subgroup_size = ret; in llvmpipe_get_compute_param() local
|
| /external/OpenCL-CTS/test_conformance/subgroups/ |
| D | subhelpers.h | 72 size_t subgroup_size; member
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | gpu_info.cc | 823 for (auto subgroup_size : supported_subgroup_sizes) { in SupportsSubGroupWithSize() local
|
| D | gpu_info.h | 309 uint32_t subgroup_size = 0; member
|
| /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
|
| /external/mesa3d/src/gallium/drivers/r600/ |
| D | r600_pipe_common.c | 1047 uint32_t *subgroup_size = ret; in r600_get_compute_param() local
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_shader.c | 377 unsigned subgroup_size, unsigned ballot_bit_size) in radv_shader_compile_to_nir()
|
| D | radv_pipeline.c | 3110 unsigned subgroup_size = 64, ballot_bit_size = 64; in radv_create_shaders() local
5425 const VkPipelineShaderStageRequiredSubgroupSizeCreateInfoEXT *subgroup_size = in radv_generate_compute_pipeline_key() local
|
| D | radv_shader.h | 110 uint8_t subgroup_size; member
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir.h | 4641 uint8_t subgroup_size; member
|
| /external/rust/crates/ash/src/vk/ |
| D | definitions.rs | 20688 pub subgroup_size: u32, field
20732 pub fn subgroup_size(mut self, subgroup_size: u32) -> Self { in subgroup_size() method
37790 pub subgroup_size: u32, field
37857 pub fn subgroup_size(mut self, subgroup_size: u32) -> Self { in subgroup_size() method
39221 pub subgroup_size: u32, field
39302 pub fn subgroup_size(mut self, subgroup_size: u32) -> Self { in subgroup_size() method
|