Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc218 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
224 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
230 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
236 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/lite/delegates/gpu/common/tasks/
Dresize.cc147 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
304 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dmean_stddev_normalization.cc211 const int grid_x = work_group_size_.x; in GetGridSize() local
Dconv_weights_converter.cc180 const int grid_x = in GetGridSize() local
Dconv_metal.cc624 int grid_x = DivideRoundUp(dst_shape.w, block_size.x); in GetGroupsCount() local
636 int grid_x = DivideRoundUp(dst_shape.w, block_size.x); in GetGroupsCountForLinearWH() local
648 int grid_x = DivideRoundUp(dst_shape.w, block_size.x); in GetGroupsCountForLinearWHS() local
960 const int grid_x = in BindArguments() local
969 int grid_x = in GetGridSize() local
Ddepthwise_conv_3x3_stride_h2.cc191 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dsplit.cc145 const int grid_x = width * batch; in GetGridSize() local
Dstrided_slice.cc170 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Ddepthwise_conv_3x3.cc285 const int grid_x = DivideRoundUp(dst_[0]->Width(), 2) * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_thin.cc151 const int grid_x = src_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_3x3_thin.cc193 const int grid_x = src_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dwinograd.cc461 const int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
675 const int grid_x = tiles_x * tiles_y * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_4x4.cc381 const int grid_x = DivideRoundUp(dst_[0]->Width() + 2, 2) * dst_[0]->Batch(); in GetGridSize() local
Dconvolution_transposed_3x3.cc365 const int grid_x = DivideRoundUp(dst_[0]->Width(), 2) * dst_[0]->Batch(); in GetGridSize() local
Dreduce.cc452 int grid_x = dst_[0]->Width() * dst_[0]->Batch(); in GetGridSize() local
Dconv_buffer_1x1.cc305 const int grid_x = in GetGridSize() local
Dconvolution_transposed.cc564 const int grid_x = DivideRoundUp(aligned_w, block_size_.x) * dst_[0]->Batch(); in GetGridSize() 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/internal/gpu/
Dcupti_collector.h91 uint32 grid_x; member
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_cs.c1279 …unsigned grid_x = (iter_idx - (grid_z * (job_info->grid_size[0] * job_info->grid_size[1])) - (grid… in cs_exec_fn() local