| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Sort.cu | 61 constexpr int block_x = sort_size / items_per_thread; in sort() local
125 const int block_x = at::cuda::warp_size(); in sort() local
|
| D | DistanceKernel.cu | 273 const int block_x = 16; in pdist_backward_kernel_impl() local
316 const int block_x = 64; in cdist_backward_kernel_impl() local
|
| D | DilatedMaxPool2d.cu | 360 int block_x = std::min<int>( in TORCH_IMPL_FUNC() local
495 int block_x = std::min<int>( in TORCH_IMPL_FUNC() local
|
| D | AdaptiveAveragePooling.cu | 505 int block_x = std::min<int>( in adaptive_avg_pool2d_out_cuda_template() local
659 int block_x = std::max<int>(std::min<int>( in adaptive_avg_pool2d_backward_out_cuda_template() local
|
| D | UpSampleBilinear2d.cu | 701 int block_x = std::min<int>(maxThreadsDim[0], at::cuda::warp_size()); in upsample_gen2d_aa_out_cuda_template() local
798 int block_x = std::min<int>(maxThreadsDim[0], at::cuda::warp_size()); in upsample_gen2d_aa_backward_out_cuda_template() local
|
| D | UpSampleNearest2d.cu | 284 int block_x = std::min<int>( in upsample_nearest2d_out_cuda_template() local
|
| D | Normalization.cuh | 144 int block_x = std::min(lastPow2(stride), OPTIMAL_TILE_W); in flexible_launch_configs() local
826 …int block_x = std::min<int>(std::max<int>(getNumThreads(feature_size), warp_size), MAX_BLOCK_SIZE/… in batch_norm_backward_reduce_cuda_template() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | fully_connected.h | 72 for (int block_x = 0; 4 * block_x < padded_src_channels; block_x++) { in RearrangeFCWeightsToIOO4I4() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.h | 53 for (int block_x = 0; 4 * block_x < padded_src_channels; block_x++) { in RearrangeFCWeightsToIOO4I4() local
|
| /external/mesa3d/src/amd/common/nir/ |
| D | ac_nir_meta_cs_blit.c | 449 unsigned block_x, unsigned block_y, unsigned block_z, in set_work_size()
1018 unsigned block_x, block_y, block_z; in ac_prepare_compute_blit() local
|
| /external/pdfium/third_party/libopenjpeg/ |
| D | sparse_array.c | 127 OPJ_UINT32 x, block_x; in opj_sparse_array_int32_read_or_write() local
|
| /external/mesa3d/src/util/ |
| D | texcompress_astc_luts.cpp | 337 for (unsigned block_x = 0; block_x < block_width; block_x++) in PartitionTable() local
|
| /external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| D | cupti_collector.h | 94 uint32 block_x; member
|
| D | rocm_tracer.h | 63 uint32_t block_x; member
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 1081 inline TensorShape compute_batch_to_space_shape(const ITensorInfo *input, const int block_x, const … in compute_batch_to_space_shape()
1165 inline TensorShape compute_space_to_batch_shape(const ITensorInfo *input, const int block_x, const … in compute_space_to_batch_shape()
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_compute_blit.c | 91 set_work_size(struct pipe_grid_info *info, unsigned block_x, unsigned block_y, unsigned block_z, in set_work_size()
|
| /external/mesa3d/src/gallium/drivers/softpipe/ |
| D | sp_setup.c | 190 block_x(int x) in block_x() function
|
| /external/libgav1/src/tile/ |
| D | tile.cc | 1017 TransformSize tx_size, int block_x, in ComputeTransformType()
|