Home
last modified time | relevance | path

Searched defs:work_group_size (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dcl_command_queue.cc63 const int3& work_group_size, in Dispatch()
88 const int3& work_group_size) { in Dispatch()
209 const int3& work_group_size) { in Dispatch()
221 const int3& work_group_size, int n, int flush_period) { in DispatchNTimes()
H A Dcl_operation.cc179 const int3& work_group_size, in RestoreDeserialized()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A Dconverter.cc63 const int3 work_group_size = work_groups[0]; in DispatchKernel() local
203 const int3 work_group_size = {16, 8, 1}; in Convert() local
/aosp_15_r20/external/deqp/external/openglcts/modules/glesext/texture_buffer/
H A DesextcTextureBufferAtomicFunctions.cpp74 glw::GLint work_group_size = 0; in initTest() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_metal_simd.h55 int3 work_group_size; member
H A Dmean_stddev_normalization.cc66 const std::string& dst_value, int3 work_group_size, in GetReduceCode()
H A Dconv_generic.h56 int3 work_group_size; member
H A Ddepthwise_conv.h81 int3 work_group_size = int3(1, 1, 1); member
H A Dreduce.cc178 const int3& work_group_size, in GetReduceKernelCode()
H A Dconv_generic.cc1209 int3 work_group_size; member
1216 const int3& work_group_size, in CreateWorkGroupSizeOption()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dworkgroup_selection.cc164 const int work_group_size = x * y * z; in GenerateWorkGroupSizes() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dutil.cc210 int3 GetWorkGroupsCount(const int3& grid_size, const int3& work_group_size) { in GetWorkGroupsCount()
H A Dgpu_operation.h132 int3 work_group_size; member
H A Dgpu_operation.cc34 const int3& work_group_size, in GetWorkGroupsCountInternal()
H A Dserialization_base_generated.h1692 const tflite::gpu::data::Int3 *work_group_size() const { in work_group_size() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dcompute_task.cc355 void ComputeTask::SetWorkGroupSize(const int3& work_group_size) { in SetWorkGroupSize()
/aosp_15_r20/external/OpenCL-CTS/test_conformance/basic/
H A Dtest_imagedim.cpp304 size_t work_group_size = 0; in test_imagedim_non_pow2() local
/aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/core/
H A Dkernel.rs323 work_group_size: [usize; 3], field
1679 pub fn work_group_size(&self) -> [usize; 3] { in work_group_size() method
/aosp_15_r20/external/deqp/external/openglcts/modules/gles31/
H A Des31cComputeShaderTests.cpp2328 const GLint work_group_size[3] = {128, 128, 64}; in Run() local
/aosp_15_r20/external/deqp/external/openglcts/modules/gl/
H A Dgl4cComputeShaderTests.cpp2339 const GLint work_group_size[3] = {1024, 1024, 64}; in Run() local