Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dwork_group_picking.cc220 int wg_xy_size = max_size / wg_z; in GetWorkGroup() local
221 int wg_x = std::min(DivideRoundUp(grid.x, 2), wg_xy_size); in GetWorkGroup()
222 int wg_y = std::min(wg_xy_size / wg_x, grid.y); in GetWorkGroup()
228 int wg_xy_size = std::min(256, max_size) / wg_z; in GetWorkGroupConv() local
229 int wg_x = std::min(grid.x, wg_xy_size); in GetWorkGroupConv()
230 int wg_y = std::min(wg_xy_size / wg_x, grid.y); in GetWorkGroupConv()