Home
last modified time | relevance | path

Searched defs:grid_x (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dresize.cc102 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
201 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dcumsum.cc93 const int grid_x = width * batch; in GetGridSize() local
Dconv_weights_converter.cc233 const int grid_x = in GetGridSize() local
Dstrided_slice.cc171 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Ddepthwise_conv_3x3_stride_h2.cc197 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_thin.cc151 const int grid_x = src_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Ddepthwise_conv_3x3.cc299 const int grid_x = DivideRoundUp(dst_[0]->Width(), 2) * dst_[0]->Batch(); in GetGridSize() local
Dsplit.cc205 const int grid_x = width * batch; in GetGridSize() local
Dwinograd.cc507 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
725 const int grid_x = tiles_x * tiles_y * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_3x3_thin.cc208 const int grid_x = src_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_4x4.cc361 const int grid_x = DivideRoundUp(dst_[0]->Width() + 2, 2) * dst_[0]->Batch(); in GetGridSize() local
Dmean_stddev_normalization.cc326 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_3x3.cc359 const int grid_x = DivideRoundUp(dst_[0]->Width(), 2) * dst_[0]->Batch(); in GetGridSize() local
Ddepthwise_conv.cc124 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dreduce.cc502 int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dconv_generic.cc1140 int grid_x = DivideRoundUp(dst_shape.w, block_size.x) * dst_shape.b; in GetGroupsCount() local
1152 int grid_x = DivideRoundUp(dst_shape.w, block_size.x) * dst_shape.b; in GetGroupsCountForLinearWH() local
1164 int grid_x = DivideRoundUp(dst_shape.w, block_size.x) * dst_shape.b; in GetGroupsCountForLinearWHS() local
Dconvolution_transposed.cc594 const int grid_x = DivideRoundUp(aligned_w, block_size_.x) * dst_[0]->Batch(); in GetGridSize() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc463 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
469 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
475 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
481 const int grid_x = dst_[0]->Batch(); in GetGridSize() local
Dwork_group_picking.cc203 int grid_x = GetMaxSizeWithMinPenalty(grid.x, 512 / grid_z); in GetWorkGroupXY128ConvLinear() local
/external/tensorflow/tensorflow/core/util/
Dgpu_launch_config.h247 int grid_x = std::min(DivUp(xdim, block_cols), max_blocks); in GetGpu2DLaunchConfig() local
/external/tensorflow/tensorflow/core/profiler/backends/gpu/
Dcupti_collector.h100 uint32 grid_x; member
Drocm_tracer.h69 uint32_t grid_x; member
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_cs.c1638 unsigned grid_z, grid_y, grid_x; in cs_exec_fn() local
2191 for (unsigned grid_x = 0; grid_x < total_grid[0]; grid_x += job_strides[0]) { in llvmpipe_draw_mesh_tasks() local