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
|