Home
last modified time | relevance | path

Searched defs:input_slice_size (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp118 int64_t input_slice_size = input_depth * input_height * input_width; in cpu_upsample_nearest_backward() local
253 int64_t input_slice_size = input_depth * input_height * input_width * channels; in cpu_upsample_nearest_backward_channels_last() local
448 int64_t input_slice_size = input_depth * input_height * input_width; in cpu_upsample_linear_backward() local
624 int64_t input_slice_size = input_depth * input_height * input_width * channels; in cpu_upsample_linear_backward_channels_last() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleBicubic2d.cpp121 auto input_slice_size = input_height * input_width; in upsample_bicubic2d_backward_out_frame() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Drfft2d.cc378 int input_slice_size = input_height * input_width; in Rfft2dHelper() local
H A Dirfft2d.cc326 int input_slice_size = input_height * input_width; in Irfft2dHelper() local