Searched defs:block_z_size (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReplicationPadding.cu | 290 int64_t block_z_size = std::min(size0 - block_z, static_cast<int64_t>(65535)); in replication_pad2d_backward_out_cuda_template() local
404 int64_t block_z_size = std::min(size0 - block_z, static_cast<int64_t>(65535)); in replication_pad3d_backward_out_cuda_template() local
459 int64_t block_z_size = std::min(size0 - block_z, static_cast<int64_t>(65535)); in TORCH_IMPL_FUNC() local
521 int64_t block_z_size = std::min(size0 - block_z, static_cast<int64_t>(65535)); in TORCH_IMPL_FUNC() local
562 int64_t block_z_size = std::min(size0 - block_z, static_cast<int64_t>(65535)); in TORCH_IMPL_FUNC() local
657 int64_t block_z_size = std::min(size0 - block_z, static_cast<int64_t>(65535)); in TORCH_IMPL_FUNC() local
|
| H A D | ReflectionPad.cu | 340 int64_t block_z_size = std::min(size_z - block_z, static_cast<int64_t>(65535)); in reflection_pad2d_out_template() local
412 int64_t block_z_size = std::min(size_z - block_z, static_cast<int64_t>(65535)); in reflection_pad2d_backward_out_template() local
609 int64_t block_z_size = std::min(size_z - block_z, static_cast<int64_t>(65535)); in TORCH_IMPL_FUNC() local
662 int64_t block_z_size = std::min(size_z - block_z, static_cast<int64_t>(65535)); in TORCH_IMPL_FUNC() local
|