Home
last modified time | relevance | path

Searched refs:work_group_launch_order (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc39 const int3& work_group_launch_order) { in GetWorkGroupsCount() argument
49 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
50 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
57 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
58 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
59 work_groups_count.z = wgs[work_group_launch_order[2]]; in GetWorkGroupsCount()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc24 const int3& work_group_launch_order) { in GetWorkGroupsCount() argument
34 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
35 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
42 work_groups_count.x = wgs[work_group_launch_order[0]]; in GetWorkGroupsCount()
43 work_groups_count.y = wgs[work_group_launch_order[1]]; in GetWorkGroupsCount()
44 work_groups_count.z = wgs[work_group_launch_order[2]]; in GetWorkGroupsCount()
Dserialization.cc821 op->work_group_launch_order_.x = fb_op->work_group_launch_order()->x(); in Decode()
822 op->work_group_launch_order_.y = fb_op->work_group_launch_order()->y(); in Decode()
823 op->work_group_launch_order_.z = fb_op->work_group_launch_order()->z(); in Decode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_3x3.cc58 int2 padding, int3 work_group_launch_order) { in GenerateConvolutionTransposedCode() argument
141 launch_remap[work_group_launch_order.x] = 0; in GenerateConvolutionTransposedCode()
142 launch_remap[work_group_launch_order.y] = 1; in GenerateConvolutionTransposedCode()
143 launch_remap[work_group_launch_order.z] = 2; in GenerateConvolutionTransposedCode()
147 if (work_group_launch_order[id] == id) { in GenerateConvolutionTransposedCode()
Dconv_metal.cc171 ids_params.launch_order = params.work_group_launch_order; in GenerateConvolution()
706 params.work_group_launch_order = int3(0, 1, 2); in GetConvParamsForA7A8()
742 params.work_group_launch_order = int3(0, 1, 2); in GetConvParamsForA7A8()
807 params.work_group_launch_order = int3(2, 0, 1); in GetConvParamsForA9AndHigher()
815 params.work_group_launch_order = int3(0, 1, 2); in GetConvParamsForA9AndHigher()
867 params.work_group_launch_order = int3(2, 0, 1); in GetConvParamsForIntel()
891 params.work_group_launch_order = int3(1, 0, 2); in GetConvParamsForIntel()
903 params.work_group_launch_order = int3(2, 0, 1); in GetConvParamsForAMD()
939 params.work_group_launch_order = int3(2, 0, 1); in GetConvParams()
1064 desc.work_group_launch_order_ = params.work_group_launch_order; in CreateConvolutionMetal()
[all …]
Dconvolution_transposed_3x3.h86 int2 padding, int3 work_group_launch_order);
Dconv_metal.h43 int3 work_group_launch_order; member
Dconv_powervr.cc74 const int3& work_group_launch_order, in GenerateBlockCoords() argument
79 launch_remap[work_group_launch_order.x] = 0; in GenerateBlockCoords()
80 launch_remap[work_group_launch_order.y] = 1; in GenerateBlockCoords()
81 launch_remap[work_group_launch_order.z] = 2; in GenerateBlockCoords()
102 if (work_group_launch_order[0] == 0) { in GenerateBlockCoords()
122 if (work_group_launch_order[1] == 1) { in GenerateBlockCoords()
131 if (work_group_launch_order[0] == 0) { in GenerateBlockCoords()
140 if (work_group_launch_order[1] == 1) { in GenerateBlockCoords()
156 if (work_group_launch_order[2] == 2) { in GenerateBlockCoords()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dserialization_base.fbs238 work_group_launch_order:Int3;
Dserialization_base_generated.h2159 const tflite::gpu::data::Int3 *work_group_launch_order() const { in work_group_launch_order() function
2204 verifier.VerifyTable(work_group_launch_order()) && in Verify()
2267 flatbuffers::Offset<tflite::gpu::data::Int3> work_group_launch_order) { in add_work_group_launch_order()
2269 work_group_launch_order); in add_work_group_launch_order()
2323 flatbuffers::Offset<tflite::gpu::data::Int3> work_group_launch_order = 0,
2341 builder_.add_work_group_launch_order(work_group_launch_order);
2368 flatbuffers::Offset<tflite::gpu::data::Int3> work_group_launch_order = 0,
2398 definition, grid_dimension, work_group_launch_order, grid_size,