Home
last modified time | relevance | path

Searched defs:grid_z (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dresize.cc104 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
203 const int grid_z = dst_[0]->Slices() * dst_[0]->Depth(); in GetGridSize() local
Dcumsum.cc95 const int grid_z = slices; in GetGridSize() local
Dconv_weights_converter.cc236 const int grid_z = weights_shape.w * weights_shape.h; in GetGridSize() local
Dstrided_slice.cc173 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Ddepthwise_conv_3x3_stride_h2.cc199 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dconvolution_transposed_thin.cc153 const int grid_z = 1; in GetGridSize() local
Ddepthwise_conv_3x3.cc301 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dsplit.cc207 const int grid_z = slices; in GetGridSize() local
Dwinograd.cc509 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
727 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dconvolution_transposed_3x3_thin.cc210 const int grid_z = 1; in GetGridSize() local
Dconvolution_transposed_4x4.cc363 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dmean_stddev_normalization.cc328 const int grid_z = work_group_size_.z; in GetGridSize() local
Dconvolution_transposed_3x3.cc361 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Ddepthwise_conv.cc126 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dreduce.cc504 int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dconv_generic.cc1142 int grid_z = DivideRoundUp(dst_slices, block_size.w); in GetGroupsCount() local
1154 int grid_z = DivideRoundUp(dst_slices, block_size.w); in GetGroupsCountForLinearWH() local
1166 int grid_z = DivideRoundUp(dst_slices, block_size.w); in GetGroupsCountForLinearWHS() local
Dconvolution_transposed.cc597 const int grid_z = DivideRoundUp(dst_[0]->Slices(), block_size_.w); in GetGridSize() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc465 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
471 const int grid_z = 1; in GetGridSize() local
477 const int grid_z = dst_[0]->Depth(); in GetGridSize() local
483 const int grid_z = 1; in GetGridSize() local
Dwork_group_picking.cc199 int grid_z = GetBiggestDividerWithPriority(grid.z, 4); in GetWorkGroupXY128ConvLinear() local
208 int grid_z = GetBiggestDividerWithPriority(grid.z, 4); in GetWorkGroupXY128Conv() local
/external/tensorflow/tensorflow/core/profiler/backends/gpu/
Dcupti_collector.h104 uint32 grid_z; member
Drocm_tracer.h73 uint32_t grid_z; member
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_cs.c1277 unsigned grid_z = iter_idx / (job_info->grid_size[0] * job_info->grid_size[1]); in cs_exec_fn() local