Home
last modified time | relevance | path

Searched refs:num_workgroups (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dserialization_test.cc43 uint3 num_workgroups; member
56 const uint3& num_workgroups, in OnProgram()
59 {parameters, objects, workgroup_size, num_workgroups, shader_index}); in OnProgram()
184 uint3 num_workgroups(10, 20, 30); in TEST() local
186 builder.AddProgram(parameters, objects, workgroup_size, num_workgroups, 1); in TEST()
193 EXPECT_EQ(num_workgroups.data_, handler.programs[0].num_workgroups.data_); in TEST()
Dapi.cc178 uint3 num_workgroups; member
204 uint3 num_workgroups = DivideRoundUp(code.workload, workgroup_size); in Add() local
220 num_workgroups, in Add()
267 program.num_workgroups)); in NewRun()
285 const uint3& num_workgroups, in OnProgram() argument
300 num_workgroups, in OnProgram()
335 program.workgroup_size, program.num_workgroups, in Serialize()
Dserialization.h51 const uint3& workgroup_size, const uint3& num_workgroups,
75 const uint3& num_workgroups,
Druntime.h50 const uint3& num_workgroups);
100 uint3 num_workgroups; member
Dserialization.cc327 const uint3& workgroup_size, const uint3& num_workgroups, in AddProgram() argument
329 Offset<data::Uint3> fb_workgroups = Encode(num_workgroups, &builder_); in AddProgram()
607 uint3 num_workgroups(program->number_workgroups()->x(), in DeserializeCompiledModel() local
611 num_workgroups, in DeserializeCompiledModel()
Druntime.cc256 const uint3& num_workgroups) { in AddProgram() argument
265 CompiledProgramDescriptor{std::move(program), num_workgroups, {}}); in AddProgram()
651 descriptor.num_workgroups)); in Execute()
Dapi2.cc601 auto num_workgroups = DivideRoundUp(code.workload, workgroup); in Build() local
603 code.objects, num_workgroups); in Build()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/converters/
Dphwc4_to_bhwc.cc86 uint3 num_workgroups = DivideRoundUp(workload, workgroup_size_); in Convert() local
97 return command_queue->Dispatch(program_, num_workgroups); in Convert()
99 return program_.Dispatch(num_workgroups); in Convert()
Dbhwc_to_phwc4.cc92 uint3 num_workgroups = DivideRoundUp(workload, workgroup_size_); in Convert() local
101 return command_queue->Dispatch(program_, num_workgroups); in Convert()
103 return program_.Dispatch(num_workgroups); in Convert()
/external/OpenCL-CTS/test_conformance/workgroups/
Dtest_wg_broadcast.cpp280 size_t num_workgroups; in test_work_group_broadcast_2D() local
313 num_workgroups = MAX(n_elems/wg_size[0], 16); in test_work_group_broadcast_2D()
314 globalsize[0] = num_workgroups * localsize[0]; in test_work_group_broadcast_2D()
315 globalsize[1] = num_workgroups * localsize[1]; in test_work_group_broadcast_2D()
407 size_t num_workgroups; in test_work_group_broadcast_3D() local
440 num_workgroups = MAX(n_elems/wg_size[0], 8); in test_work_group_broadcast_3D()
441 globalsize[0] = num_workgroups * localsize[0]; in test_work_group_broadcast_3D()
442 globalsize[1] = num_workgroups * localsize[1]; in test_work_group_broadcast_3D()
443 globalsize[2] = num_workgroups * localsize[2]; in test_work_group_broadcast_3D()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter.cc67 uint3 num_workgroups = DivideRoundUp(workload, workgroup_size_); in Dispatch() local
69 return command_queue_->Dispatch(program_, num_workgroups); in Dispatch()
71 return program_.Dispatch(num_workgroups); in Dispatch()
/external/mesa3d/src/intel/vulkan/
DgenX_cmd_buffer.c2642 cmd_buffer->state.compute.num_workgroups, in emit_binding_table()
2649 cmd_buffer->state.compute.num_workgroups); in emit_binding_table()
4591 cmd_buffer->state.compute.num_workgroups = (struct anv_address) { in genX()
4637 cmd_buffer->state.compute.num_workgroups = addr; in genX()
Danv_private.h2919 struct anv_address num_workgroups; member
/external/mesa3d/docs/relnotes/
D20.0.0.rst696 - aco: Make num_workgroups and local_invocation_ids one argument each