Home
last modified time | relevance | path

Searched defs:input_w (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReflectionPad.cpp45 int64_t input_w = input.size(dim_w); in TORCH_META_FUNC() local
85 int64_t input_w = input.size(dim_w); in TORCH_META_FUNC() local
131 int64_t input_w = input.size(dim_w); in TORCH_META_FUNC() local
192 int64_t input_w = input.size(dim_w); in TORCH_META_FUNC() local
237 int64_t input_w = input.size(dim_w); in reflection_pad2d_out_template() local
287 int64_t input_w = input.size(dim_w); in reflection_pad2d_backward_out_template() local
H A DRNN.cpp881 auto input_w = params.first.linear_ih(input); in operator ()() local
941 Tensor input_w; in operator ()() local
1000 Tensor input_w; in operator ()() 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()
98 int64_t input_w, in reflection_pad1d_out_kernel()
112 int64_t input_w, in reflection_pad1d_backward_out_kernel()
298 int input_w = input_.size(dim_w); in reflection_pad2d_out_template() local
388 int input_w = input.size(dim_w); in reflection_pad2d_backward_out_template() local
454 int64_t input_w = input_.size(dim_w); in TORCH_IMPL_FUNC() local
511 int64_t input_w = input.size(dim_w); in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/armnn/third-party/stb/
H A Dstb_image_resize.h497 int input_w; member
1253 int input_w = stbir_info->input_w; in stbir__decode_scanline() local
1530 int input_w = stbir_info->input_w; in stbir__resample_horizontal_downsample() 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()
2428 const void* input_data, int input_w, int input_h, int input_stride_in_bytes, in stbir__resize_arbitrary()
2460 STBIRDEF int stbir_resize_uint8( const unsigned char *input_pixels , int input_w , int input_h … in stbir_resize_uint8()
2470 STBIRDEF int stbir_resize_float( const float *input_pixels , int input_w , int input_h , int in… in stbir_resize_float()
2480 STBIRDEF int stbir_resize_uint8_srgb(const unsigned char *input_pixels , int input_w , int input_h … in stbir_resize_uint8_srgb()
2490 STBIRDEF int stbir_resize_uint8_srgb_edgemode(const unsigned char *input_pixels , int input_w , int… in stbir_resize_uint8_srgb_edgemode()
2501 STBIRDEF int stbir_resize_uint8_generic( const unsigned char *input_pixels , int input_w , int inpu… in stbir_resize_uint8_generic()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuIm2ColKernel.cpp91 int input_w, in linearize_volume_nchw()
188 int input_w, in linearize_volume_nhwc()
261 const int input_w = src->info()->dimension(width_idx); in run_im2col() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_input_ops.h122 auto input_w = GetTensorDimIndex(data_format, 'W'); in operator() local
H A Dconv_grad_input_ops.cc86 auto input_w = GetTensorDimIndex(data_format, 'W'); in operator ()() local
/aosp_15_r20/packages/apps/LegacyCamera/jni/feature_stab/src/dbreg/
Ddbreg.cpp747 int input_w = m_im_width*2; in GenerateQuarterResImage() local
/aosp_15_r20/packages/apps/LegacyCamera/jni/
Dfeature_mos_jni.cpp121 void GenerateQuarterResImagePlanar(ImageType im, int input_w, int input_h, in GenerateQuarterResImagePlanar()
/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dvulkan_compute_api_test.cpp1816 int64_t input_w = 256; in TEST() local