Searched defs:nplane (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| 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()
296 int nplane = input_.size(plane_dim); in reflection_pad2d_out_template() local
386 int nplane = input.size(plane_dim); in reflection_pad2d_backward_out_template() local
453 int64_t nplane = input_.size(dim_plane); in TORCH_IMPL_FUNC() local
510 int64_t nplane = input.size(dim_plane); in TORCH_IMPL_FUNC() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ReflectionPad.cpp | 44 int64_t nplane = input.size(dim_plane); in TORCH_META_FUNC() local
128 int64_t nplane = input.size(dim_plane); in TORCH_META_FUNC() local
235 int64_t nplane = input.size(dim_slices); in reflection_pad2d_out_template() local
|