Home
last modified time | relevance | path

Searched defs:nbatch (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DPixelShuffleKernel.cpp29 int64_t nbatch = numel / (channels * height * width); in cpu_pixel_shuffle() local
65 int64_t nbatch = input.size(0); in cpu_pixel_shuffle_channels_last() local
127 int64_t nbatch = numel / (channels * height * width); in cpu_pixel_unshuffle() local
164 int64_t nbatch = input.size(0); in cpu_pixel_unshuffle_channels_last() local
H A DAdaptiveAvgPoolKernel.cpp83 int64_t nbatch = input.size(0); in cpu_adaptive_avg_pool2d_channels_last() local
170 int64_t nbatch = input.size(0); in cpu_adaptive_avg_pool2d_channels_last() local
317 int64_t nbatch = grad_input.size(0); in cpu_adaptive_avg_pool2d_backward_channels_last() local
489 int64_t nbatch = input.size(0); in cpu_adaptive_avg_pool3d_channels_last() local
585 int64_t nbatch = input.size(0); in cpu_adaptive_avg_pool3d_channels_last() local
751 int64_t nbatch = grad_input.size(0); in cpu_adaptive_avg_pool3d_backward_channels_last() local
H A DAdaptiveMaxPoolKernel.cpp102 int64_t nbatch = input.size(0); in cpu_adaptive_max_pool2d_channels_last() local
219 int64_t nbatch = input.size(0); in cpu_adaptive_max_pool2d_channels_last() local
403 int64_t nbatch = grad_input.size(0); in cpu_adaptive_max_pool2d_backward_channels_last() local
576 int64_t nbatch = input.size(0); in cpu_adaptive_max_pool3d_channels_last() local
701 int64_t nbatch = input.size(0); in cpu_adaptive_max_pool3d_channels_last() local
897 int64_t nbatch = grad_input.size(0); in cpu_adaptive_max_pool3d_backward_channels_last() local
H A DChannelShuffleKernel.cpp23 int64_t nbatch = input.size(0); in cpu_channel_shuffle() local
68 int64_t nbatch = input.size(0); in cpu_channel_shuffle_cl() local
H A DAvgPoolKernel.cpp121 int64_t nbatch = input.size(0); in cpu_avg_pool2d_channels_last() local
235 int64_t nbatch = input.size(0); in cpu_avg_pool2d_channels_last() local
432 int64_t nbatch = grad_input.size(0); in cpu_avg_pool2d_backward_channels_last() local
663 int64_t nbatch = input.size(0); in cpu_avg_pool3d_channels_last() local
786 int64_t nbatch = input.size(0); in cpu_avg_pool3d_channels_last() local
1001 int64_t nbatch = grad_input.size(0); in cpu_avg_pool3d_backward_channels_last() local
H A DPaddingKernel.cpp17 int64_t nbatch; member
249 int64_t nbatch = p.nbatch; in cpu_padding_channels_last() local
411 int64_t nbatch = p.nbatch; in cpu_padding_backward_channels_last() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h119 int64_t nbatch = input_size[0]; in upsample_1d_common_check() local
148 int64_t nbatch = input_size[0]; in upsample_2d_common_check() local
186 int64_t nbatch = input_size[0]; in upsample_3d_common_check() local
216 int64_t nbatch, in upsample_2d_shape_check()
H A DReplicationPadding.cpp32 int64_t nbatch = 1; in TORCH_META_FUNC() local
97 int64_t nbatch = 1; in TORCH_META_FUNC() local
140 int64_t nbatch = 1; in TORCH_META_FUNC() local
H A DAveragePool2d.cpp44 const int64_t nbatch = input.ndimension() == 4 ? input.size(-4) : 1; in TORCH_PRECOMPUTE_META_FUNC() local
126 const int64_t nbatch = input.ndimension() == 4 ? input.size(-4) : 1; in TORCH_META_FUNC() local
H A DAveragePool3d.cpp57 const int64_t nbatch = input.size(0); in TORCH_META_FUNC() local
311 const int64_t nbatch = input.size(0); in TORCH_IMPL_FUNC() local
488 const int64_t nbatch = input.size(0); in TORCH_IMPL_FUNC() local
H A DUpSampleBicubic2d.cpp115 int64_t nbatch, in upsample_bicubic2d_backward_out_frame()
194 int64_t nbatch = input_size[0]; in upsample_bicubic2d_backward_kernel() local
H A DReflectionPad.cpp30 int64_t nbatch = 1; in TORCH_META_FUNC() local
217 int64_t nbatch = 1; in reflection_pad2d_out_template() local
H A DAdaptiveAveragePooling.cpp50 int64_t nbatch = input.size(0); in adaptive_avg_pool2d_out_cpu_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DUpSampleNearest2d.cpp37 int64_t nbatch, in upsample_nearest2d_out_frame()
93 int64_t nbatch, in upsample_nearest2d_out_frame_nhwc()
139 int64_t nbatch = input.size(0); in _upsample_nearest2d_quantized_cpu() local
H A DUpSampleNearest3d.cpp37 int64_t nbatch, in upsample_nearest3d_out_frame()
94 int64_t nbatch, in upsample_nearest3d_out_frame_nhwc()
153 int64_t nbatch = input.size(0); in _upsample_nearest3d_quantized_cpu() local
H A DAveragePool3d.cpp78 const int64_t nbatch = input_.ndimension() == 5 ? input_.size(-5) : 1; in get_output_shape() local
108 const int64_t nbatch = input.ndimension() == 5 ? input.size(-5) : 1; in q_avg_pool3d() local
H A DUpSampleBilinear2d.cpp46 int64_t nbatch, in upsample_bilinear2d_out_frame()
164 int64_t nbatch = input.size(0); in upsample_bilinear2d_quantized_cpu() local
H A DPooling.cpp104 int64_t nbatch, in spatial_dilated_max_pooling3d()
216 int nbatch = 1; in q_maxpool_2d() local
388 int nbatch = qx.size(0); in q_maxpool_3d() local
H A DAveragePool2d.cpp166 const int64_t nbatch = input_.ndimension() == 4 ? input_.size(-4) : 1; in get_output_shape() local
194 const int64_t nbatch = input.ndimension() == 4 ? input.size(-4) : 1; in q_avg_pool2d() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu280 int nbatch = 1; in reflection_pad2d_out_template() local
372 int nbatch = 1; in reflection_pad2d_backward_out_template() local
442 int64_t nbatch = 1; in TORCH_IMPL_FUNC() local
499 int64_t nbatch = 1; in TORCH_IMPL_FUNC() local
H A DUpSampleNearest1d.cu113 int nbatch = input_.size(0); in upsample_nearest1d_out_cuda_template() local
161 int nbatch = input_size[0]; in upsample_nearest1d_backward_out_cuda_template() local
H A DUpSampleNearest3d.cu166 int nbatch = input_.size(0); in upsample_nearest3d_out_cuda_template() local
240 int nbatch = input_size[0]; in upsample_nearest3d_backward_out_cuda_template() local
H A Dim2col.cuh248 const int64_t nbatch, in col2im_batched_kernel()
296 const int64_t nbatch, in col2im_batched()
H A DUpSampleNearest2d.cu222 int nbatch = input_.size(0); in upsample_nearest2d_out_cuda_template() local
353 int nbatch = input_size[0]; in upsample_nearest2d_backward_out_cuda_template() local
H A DAveragePool3d.cu382 const int64_t nbatch = input.ndimension() == 5 ? input.size(-5) : 1; in TORCH_IMPL_FUNC() local
493 const int64_t nbatch = input.ndimension() == 5 ? input.size(-5) : 1; in TORCH_IMPL_FUNC() local

12