Searched defs:z_shift (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReflectionPad.cu | 63 int64_t output_xy, int y_shift, int z_shift, int nplane) { in get_index_mapping2d()
128 int pad_t, int pad_b, int pad_l, int pad_r, int y_shift, int z_shift, int nplane) { in reflection_pad2d_out_kernel()
148 int pad_t, int pad_b, int pad_l, int pad_r, int y_shift, int z_shift, int nplane) { in reflection_pad2d_backward_out_kernel()
171 int64_t z_shift, in parallel_reflection_pad3d()
217 int64_t y_shift, int64_t z_shift in reflection_pad3d_out_kernel()
246 int64_t y_shift, int64_t z_shift in reflection_pad3d_backward_out_kernel()
|
| H A D | ReplicationPadding.cu | 46 const int z_shift) { in replication_pad_forward_kernel1d()
70 const int z_shift) { in replication_pad_backward_kernel()
95 const int z_shift) { in replication_pad_forward_kernel2d()
124 const int z_shift) { in replication_pad_backward_kernel()
154 const int z_shift) { in replication_pad_forward_kernel3d()
193 const int z_shift) { in replication_pad_backward_kernel()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/ |
| H A D | lp_bld_depth.c | 840 unsigned z_shift = 0, z_width = 0, z_mask = 0; in lp_build_depth_stencil_test() local
|