Home
last modified time | relevance | path

Searched defs:width_scale (Results 1 – 25 of 33) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleNearest2d.cu50 float width_scale) { in upsample_nearest2d_out_frame()
93 float width_scale, in upsample_nearest2d_nhwc_out_frame()
124 float width_scale) { in upsample_nearest2d_backward_out_frame()
175 const float width_scale, in upsample_nearest2d_backward_nhwc_out_frame()
228 const float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest2d_out_cuda_template() local
359 …const float width_scale = compute_scales_value_backwards<float>(scales_w, output_width, input_widt… in upsample_nearest2d_backward_out_cuda_template() local
H A DUpSampleNearest3d.cu56 float width_scale) { in upsample_nearest3d_out_frame()
101 float width_scale) { in upsample_nearest3d_backward_out_frame()
196 const float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest3d_out_cuda_template() local
272 … float width_scale = compute_scales_value_backwards<float>(scales_w, output_width, input_width); in upsample_nearest3d_backward_out_cuda_template() local
H A DUpSampleBilinear2d.cu476 const accscalar_t width_scale, in upsample_gen2d_aa_out_frame()
570 const accscalar_t width_scale, in upsample_gen2d_aa_backward_out_frame()
714 const accscalar_t width_scale = area_pixel_compute_scale<accscalar_t>( in upsample_gen2d_aa_out_cuda_template() local
818 const accscalar_t width_scale = area_pixel_compute_scale<accscalar_t>( in upsample_gen2d_aa_backward_out_cuda_template() local
H A DUpSampleBicubic2d.cu27 const accscalar_t width_scale, in upsample_bicubic2d_out_frame()
102 const accscalar_t width_scale, in upsample_bicubic2d_backward_out_frame()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dresize_bilinear_op_gpu.cu.cc38 float height_scale, float width_scale, int batch, int in_height, in ResizeBilinearKernel_faster()
118 float width_scale, int batch, int in_height, int in_width, int channels, in ResizeBilinearKernel()
168 float height_scale, float width_scale, in ResizeBilinearGradKernel()
233 float height_scale, float inverse_height_scale, float width_scale, in ResizeBilinearDeterministicGradKernel()
286 float width_scale, int batch, int in_height, int in_width, int channels, in LegacyResizeBilinearKernel()
336 float height_scale, float width_scale, int batch, int original_height, in LegacyResizeBilinearGradKernel()
H A Dresize_nearest_neighbor_op_gpu.cu.cc38 const int out_width, const float height_scale, const float width_scale, in ResizeNearestNeighborNHWC()
69 const int out_width, const float height_scale, const float width_scale, in LegacyResizeNearestNeighborNHWC()
98 const int out_width, const float height_scale, const float width_scale, in ResizeNearestNeighborBackwardNHWC()
129 const int out_width, const float height_scale, const float width_scale, in LegacyResizeNearestNeighborBackwardNHWC()
H A Dcrop_and_resize_op.cc255 const float width_scale = in operator ()() local
492 const float width_scale = in operator ()() local
711 const float width_scale = (crop_width > 1) ? (x2 - x1) * width_ratio : 0; in operator ()() local
H A Dcrop_and_resize_op_gpu.cu.cc173 const float width_scale = in CropAndResizeBackpropImageKernel() local
276 const float width_scale = (crop_width > 1) ? (x2 - x1) * width_ratio : 0; in CropAndResizeBackpropBoxesKernel() local
H A Dresize_area_op_test.cc77 const float width_scale = in_width / static_cast<float>(out_width); in ResizeAreaBaseline() local
H A Dresize_nearest_neighbor_op.cc274 const float width_scale = in Compute() local
H A Dresize_bicubic_op_test.cc128 const float width_scale = in_width / static_cast<float>(out_width); in ResizeBicubicBaseline() local
H A Dresize_bilinear_op_test.cc96 const float width_scale = in_width / static_cast<float>(out_width); in ResizeBilinearBaseline() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp462 const opmath_t width_scale = area_pixel_compute_scale<opmath_t>( in cpu_upsample_linear_backward() local
497 const opmath_t width_scale = area_pixel_compute_scale<opmath_t>( in cpu_upsample_linear_backward() local
540 const opmath_t width_scale = area_pixel_compute_scale<opmath_t>( in cpu_upsample_linear_backward() local
640 const opmath_t width_scale = area_pixel_compute_scale<opmath_t>( in cpu_upsample_linear_backward_channels_last() local
689 const opmath_t width_scale = area_pixel_compute_scale<opmath_t>( in cpu_upsample_linear_backward_channels_last() local
H A DUpSampleKernel.cpp601 const auto width_scale = area_pixel_compute_scale<opmath_t>( in cpu_upsample_linear_channels_last() local
649 const auto width_scale = area_pixel_compute_scale<opmath_t>( in cpu_upsample_linear_channels_last() local
1954 const scalar_t width_scale = area_pixel_compute_scale<scalar_t>( in cpu_upsample_genNd_backward_aa() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_resize_bilinear_op.cc473 const float height_scale, const float width_scale, in ResizeImageReference()
522 const float height_scale, const float width_scale, in ResizeImage()
536 const float height_scale, const float width_scale, in ResizeImage()
599 const float height_scale, const float width_scale, in ResizeImage()
661 const float height_scale, const float width_scale, in ResizeBilinear()
H A Dquantized_resize_bilinear_op_test.cc116 const float height_scale, const float width_scale, in CalcReferenceResizedVal()
157 const float width_scale = in CheckTensorValue() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dimage_resizer_state.h164 float width_scale; member
249 float width_scale; member
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c360 const float width_scale = in xnn_indirection_init_resize_bilinear2d_hwc_f16() local
456 const float width_scale = in xnn_indirection_init_resize_bilinear2d_hwc_f32() local
551 const float width_scale = in xnn_indirection_init_resize_bilinear2d_hwc_q11() local
646 const float width_scale = in xnn_indirection_init_resize_bilinear2d_chw_f16() local
747 const float width_scale = in xnn_indirection_init_resize_bilinear2d_chw_f32() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DUpSampleNearest2d.cpp42 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest2d_out_frame() local
98 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest2d_out_frame_nhwc() local
H A DUpSampleNearest3d.cpp44 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest3d_out_frame() local
101 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest3d_out_frame_nhwc() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dresize_bilinear.h1514 float width_scale, const RuntimeShape& input_shape, const float* input_data, in ResizeBilinearGeneric()
1567 float width_scale, const RuntimeShape& input_shape, const T* input_data, in ResizeBilinearGenericSmallChannel()
1643 float width_scale = static_cast<float>(input_width) / output_width; in ResizeBilinear() local
1707 float width_scale = in ResizeBilinear() local
/aosp_15_r20/external/mesa3d/src/vulkan/util/
H A Dvk_format.h268 const uint8_t width_scale = ycbcr_info ? in vk_format_get_plane_width() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dresize_bilinear.h81 float width_scale = static_cast<float>(input_width) / output_width; in ResizeBilinear() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleBicubic2d.cpp127 const opmath_t width_scale = area_pixel_compute_scale<opmath_t>( in upsample_bicubic2d_backward_out_frame() local
/aosp_15_r20/external/XNNPACK/test/
H A Dresize-bilinear-operator-tester.h90 inline float width_scale() const { in width_scale() function

12