Home
last modified time | relevance | path

Searched defs:workgroup_size (Results 1 – 25 of 38) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
H A Dapi.cc180 uint3 workgroup_size; member
206 uint3 workgroup_size = workgroup_calculator.Calculate(code); in Add() local
232 const uint3& workgroup_size, size_t* size) { in AddFullShader()
287 const uint3& workgroup_size, in OnProgram()
H A Dserialization_test.cc43 uint3 workgroup_size; member
186 uint3 workgroup_size(1, 2, 3); in TEST() local
H A Dserialization.cc331 const uint3& workgroup_size, const uint3& num_workgroups, in AddProgram()
608 uint3 workgroup_size(program->workgroup_size()->x(), in DeserializeCompiledModel() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/converters/
H A Dbhwc_to_phwc4.h44 explicit ConverterBhwcToPhwc4(GlProgram program, const uint3& workgroup_size) in ConverterBhwcToPhwc4()
H A Dphwc4_to_bhwc.h44 explicit ConverterPhwc4ToBhwc(GlProgram program, const uint3& workgroup_size) in ConverterPhwc4ToBhwc()
H A Dbhwc_to_phwc4.cc35 uint3 workgroup_size = uint3(4, 4, 4); in Create() local
H A Dphwc4_to_bhwc.cc35 uint3 workgroup_size = uint3(4, 4, 4); in Create() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/
H A Dcalculator.cc65 uint3 workgroup_size = shader_code.recommended_workgroup; in Calculate() local
/aosp_15_r20/external/mesa3d/src/amd/common/
H A Dac_shader_util.c1110 unsigned workgroup_size = MAX2(es_verts, gs_inst_prims); in ac_compute_esgs_workgroup_size() local
1130 unsigned workgroup_size = MAX4(max_vtx_in, max_vtx_out, max_prim_in, max_prim_out); in ac_compute_ngg_workgroup_size() local
H A Dac_nir_meta.h182 unsigned workgroup_size; member
/aosp_15_r20/external/skia/third_party/vello/src/
H A Dshaders.rs16 pub fn workgroup_size(&self) -> ffi::WorkgroupSize { in workgroup_size() method
H A Dlib.rs201 fn workgroup_size(self: &Shader) -> WorkgroupSize; in workgroup_size() function
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_simd_selection.cpp99 const unsigned workgroup_size = cs_prog_data->local_size[0] * in elk_simd_should_compile() local
H A Delk_nir_lower_cs_intrinsics.c315 ASSERTED unsigned workgroup_size = in elk_nir_lower_cs_intrinsics() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_simd_selection.cpp109 const unsigned workgroup_size = cs_prog_data->local_size[0] * in brw_simd_should_compile() local
H A Dbrw_nir_lower_cs_intrinsics.c343 ASSERTED unsigned workgroup_size = in brw_nir_lower_cs_intrinsics() local
/aosp_15_r20/external/OpenCL-CTS/test_conformance/subgroups/
H A Dsubhelpers.cpp200 int subgroup_size, int &workgroup_size, in set_last_workgroup_params()
/aosp_15_r20/external/mesa3d/src/asahi/compiler/
H A Dagx_nir_lower_subgroups.c106 nir_def *workgroup_size = nir_load_workgroup_size(b); in lower() local
/aosp_15_r20/external/mesa3d/src/amd/vulkan/nir/
H A Dradv_nir_lower_ray_queries.c232 uint32_t workgroup_size = in init_ray_query_vars() local
608 uint32_t workgroup_size = in lower_rq_proceed() local
H A Dradv_nir_opt_tid_function.c91 unsigned workgroup_size = in update_fotid_intrinsic() local
/aosp_15_r20/external/mesa3d/src/imagination/vulkan/
H A Dpvr_query_compute.c316 const uint32_t workgroup_size[PVR_WORKGROUP_DIMENSIONS] = { in pvr_write_private_compute_dispatch() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Dconverter.cc61 absl::Status InitializeProgram(const uint3& workgroup_size, in InitializeProgram()
/aosp_15_r20/external/mesa3d/src/amd/compiler/
H A Daco_live_var_analysis.cpp371 unsigned workgroup_size = in calc_waves_per_workgroup() local
H A Daco_shader_info.h104 unsigned workgroup_size; member
/aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/mesa/compiler/
H A Dnir.rs315 pub fn workgroup_size(&self) -> [u16; 3] { in workgroup_size() method

12