Home
last modified time | relevance | path

Searched refs:launch_remap (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_4x4.cc155 int3 launch_remap; in GenerateConvolutionTransposedCode() local
156 launch_remap[work_group_launch_order_.x] = 0; in GenerateConvolutionTransposedCode()
157 launch_remap[work_group_launch_order_.y] = 1; in GenerateConvolutionTransposedCode()
158 launch_remap[work_group_launch_order_.z] = 2; in GenerateConvolutionTransposedCode()
162 grid_coords[0] = "(GROUP_ID_" + std::to_string(launch_remap[0]) + in GenerateConvolutionTransposedCode()
168 grid_coords[1] = "(GROUP_ID_" + std::to_string(launch_remap[1]) + in GenerateConvolutionTransposedCode()
174 grid_coords[2] = "(GROUP_ID_" + std::to_string(launch_remap[2]) + in GenerateConvolutionTransposedCode()
Dconvolution_transposed_3x3.cc140 int3 launch_remap; in GenerateConvolutionTransposedCode() local
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()
150 return "GROUP_ID_" + std::to_string(launch_remap[id]) + " * GROUP_SIZE_" + in GenerateConvolutionTransposedCode()
Dconv_metal.cc69 int3 launch_remap; in GlobalIdsGen() local
70 launch_remap[params.launch_order.x] = 0; in GlobalIdsGen()
71 launch_remap[params.launch_order.y] = 1; in GlobalIdsGen()
72 launch_remap[params.launch_order.z] = 2; in GlobalIdsGen()
86 c += " int linear_wh = " + params.group_ids[launch_remap.x] + " * " + in GlobalIdsGen()
97 c += " int Z = (" + params.group_ids[launch_remap.y] + " * " + in GlobalIdsGen()
106 c += " int X = (" + params.group_ids[launch_remap.x] + " * " + in GlobalIdsGen()
114 c += " int Y = (" + params.group_ids[launch_remap.y] + " * " + in GlobalIdsGen()
122 c += " int Z = (" + params.group_ids[launch_remap.z] + " * " + in GlobalIdsGen()
Dconv_powervr.cc78 int3 launch_remap; in GenerateBlockCoords() local
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()
106 std::to_string(launch_remap[0]) + " * GROUP_SIZE_0 + LOCAL_ID_0;\n"; in GenerateBlockCoords()
126 c += " int DST_S = (GROUP_ID_" + std::to_string(launch_remap[1]) + in GenerateBlockCoords()
135 c += " int DST_X = (GROUP_ID_" + std::to_string(launch_remap[0]) + in GenerateBlockCoords()
143 global_id_1 = "(GROUP_ID_" + std::to_string(launch_remap[1]) + in GenerateBlockCoords()
160 c += " int DST_S = (GROUP_ID_" + std::to_string(launch_remap[2]) + in GenerateBlockCoords()