Home
last modified time | relevance | path

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 DReplicationPadding.cu290 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 DReflectionPad.cu340 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