Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc220 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
226 const int grid_z = 1; in GetGridSize() local
232 const int grid_z = dst_[0]->Depth(); in GetGridSize() local
238 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/lite/delegates/gpu/common/tasks/
Dresize.cc149 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
306 const int grid_z = dst_[0]->Slices() * dst_[0]->Depth(); in GetGridSize() local
Dmean_stddev_normalization.cc213 const int grid_z = 1; in GetGridSize() local
Dconv_weights_converter.cc183 const int grid_z = src_[0]->Width() * src_[0]->Height(); in GetGridSize() local
Ddepthwise_conv_3x3_stride_h2.cc193 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dsplit.cc147 const int grid_z = slices; in GetGridSize() local
Dstrided_slice.cc172 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Ddepthwise_conv_3x3.cc287 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dconvolution_transposed_thin.cc153 const int grid_z = 1; in GetGridSize() local
Dconv_metal.cc626 int grid_z = DivideRoundUp(dst_slices, block_size.z); in GetGroupsCount() local
638 int grid_z = DivideRoundUp(dst_slices, block_size.z); in GetGroupsCountForLinearWH() local
650 int grid_z = DivideRoundUp(dst_slices, block_size.z); in GetGroupsCountForLinearWHS() local
972 int grid_z = DivideRoundUp(dst_[0]->Slices(), params_.block_size.z); in GetGridSize() local
Dconvolution_transposed_3x3_thin.cc195 const int grid_z = 1; in GetGridSize() local
Dwinograd.cc463 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
677 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dconvolution_transposed_4x4.cc383 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dconvolution_transposed_3x3.cc367 const int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dreduce.cc454 int grid_z = dst_[0]->Slices(); in GetGridSize() local
Dconv_buffer_1x1.cc309 const int grid_z = in GetGridSize() local
Dconvolution_transposed.cc567 const int grid_z = DivideRoundUp(dst_[0]->Slices(), block_size_.w); in GetGridSize() local
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.h95 uint32 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