Home
last modified time | relevance | path

Searched defs:output_slice_size (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleBicubic2d.cpp122 auto output_slice_size = output_height * output_width; in upsample_bicubic2d_backward_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp117 int64_t output_slice_size = output_depth * output_height * output_width; in cpu_upsample_nearest_backward() local
449 int64_t output_slice_size = output_depth * output_height * output_width; in cpu_upsample_linear_backward() local
H A DUpSampleKernel.cpp594 int64_t output_slice_size = output_depth * output_height * output_width * channels; in cpu_upsample_linear_channels_last() local
1948 int64_t output_slice_size = output_depth * output_height * output_width; in cpu_upsample_genNd_backward_aa() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Drfft2d.cc379 int output_slice_size = fft_height * (fft_width / 2 + 1); in Rfft2dHelper() local
H A Dirfft2d.cc327 int output_slice_size = fft_height * fft_width; in Irfft2dHelper() local