Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReflectionPad.cpp46 int64_t output_w = input_w + pad_l + pad_r; in TORCH_META_FUNC() local
86 int64_t output_w = input_w + pad_l + pad_r; in TORCH_META_FUNC() local
134 int64_t output_w = input_w + pad_left + pad_right; in TORCH_META_FUNC() local
195 int64_t output_w = input_w + pad_left + pad_right; in TORCH_META_FUNC() local
239 int64_t output_w = input_w + pad_l + pad_r; in reflection_pad2d_out_template() local
289 int64_t output_w = input_w + pad_l + pad_r; in reflection_pad2d_backward_out_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu35 int64_t input_w, int64_t output_w, in get_index_mapping1d()
101 auto output_w = input_w + pad_l + pad_r; in reflection_pad1d_out_kernel() local
115 auto output_w = input_w + pad_l + pad_r; in reflection_pad1d_backward_out_kernel() local
311 int output_w = input_w + pad_l + pad_r; in reflection_pad2d_out_template() local
391 int output_w = input_w + pad_l + pad_r; in reflection_pad2d_backward_out_template() local
455 int64_t output_w = input_w + pad_l + pad_r; in TORCH_IMPL_FUNC() local
512 int64_t output_w = input_w + pad_l + pad_r; in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/armnn/third-party/stb/
H A Dstb_image_resize.h502 int output_w; member
1447 int output_w = stbir_info->output_w; in stbir__resample_horizontal_upsample() local
1872 int output_w = stbir_info->output_w; in stbir__resample_vertical_upsample() local
1989 int output_w = stbir_info->output_w; in stbir__resample_vertical_downsample() local
2126 int output_w = stbir_info->output_w; in stbir__empty_ring_buffer() local
2205 static void stbir__setup(stbir__info *info, int input_w, int input_h, int output_w, int output_h, i… in stbir__setup()
2429 void* output_data, int output_w, int output_h, int output_stride_in_bytes, in stbir__resize_arbitrary()
2461 … unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_uint8()
2471 … float *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_float()
2481 … unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_uint8_srgb()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dutil.cpp646 uint64_t output_w = in computeFlops() local
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h771 const unsigned int output_w = m0 * kw * kh; in compute_indirect_buffer_shape() local