Home
last modified time | relevance | path

Searched defs:num_workgroups (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
H A Dapi.cc181 uint3 num_workgroups; member
207 uint3 num_workgroups = DivideRoundUp(code.workload, workgroup_size); in Add() local
288 const uint3& num_workgroups, in OnProgram()
H A Dserialization_test.cc44 uint3 num_workgroups; member
185 uint3 num_workgroups(10, 20, 30); in TEST() local
H A Dserialization.cc331 const uint3& workgroup_size, const uint3& num_workgroups, in AddProgram()
611 uint3 num_workgroups(program->number_workgroups()->x(), in DeserializeCompiledModel() local
H A Druntime.h101 uint3 num_workgroups; member
H A Druntime.cc259 const uint3& num_workgroups) { in AddProgram()
H A Dapi2.cc604 auto num_workgroups = DivideRoundUp(code.workload, workgroup); in Build() local
/aosp_15_r20/external/OpenCL-CTS/test_conformance/workgroups/
H A Dtest_wg_broadcast.cpp277 size_t num_workgroups; in test_work_group_broadcast_2D() local
400 size_t num_workgroups; in test_work_group_broadcast_3D() local
/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_lower_system_values.c57 nir_def *num_workgroups = nir_load_num_workgroups(b); in build_global_group_size() local
744 nir_def *num_workgroups = nir_load_num_workgroups(b); in lower_compute_system_value_instr() local
H A Dnir.h6054 uint32_t num_workgroups[3]; /* Compile-time-known dispatch sizes, or 0 if unknown. */ member
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/converters/
H A Dbhwc_to_phwc4.cc92 uint3 num_workgroups = DivideRoundUp(workload, workgroup_size_); in Convert() local
H A Dphwc4_to_bhwc.cc86 uint3 num_workgroups = DivideRoundUp(workload, workgroup_size_); in Convert() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Dconverter.cc72 uint3 num_workgroups = DivideRoundUp(workload, workgroup_size_); in Dispatch() local
/aosp_15_r20/external/mesa3d/src/amd/compiler/
H A Daco_live_var_analysis.cpp468 unsigned num_workgroups = waves * num_simd / waves_per_workgroup; in max_suitable_waves() local
/aosp_15_r20/external/mesa3d/src/amd/common/
H A Dac_nir_meta.h124 unsigned num_workgroups[3]; /* DISPATCH_DIRECT parameters */ member
/aosp_15_r20/external/mesa3d/src/gallium/drivers/d3d12/
H A Dd3d12_nir_passes.c162 nir_variable *num_workgroups; member
/aosp_15_r20/external/mesa3d/src/intel/vulkan_hasvk/
H A Danv_private.h2473 struct anv_address num_workgroups; member
/aosp_15_r20/external/mesa3d/src/intel/vulkan/
H A Danv_private.h3899 struct anv_address num_workgroups; member