| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AffineGridGenerator.cpp | 19 bool align_corners) { in linspace_from_neg_one()
36 bool align_corners) { in make_base_grid_4D()
53 bool align_corners) { in make_base_grid_5D()
70 bool align_corners) { in affine_grid_generator_4D()
83 bool align_corners) { in affine_grid_generator_5D()
89 Tensor affine_grid_generator(const Tensor& theta, IntArrayRef size, bool align_corners) { in affine_grid_generator()
108 bool align_corners) { in affine_grid_generator_4D_backward()
124 bool align_corners) { in affine_grid_generator_5D_backward()
133 Tensor affine_grid_generator_backward(const Tensor& grad, IntArrayRef size, bool align_corners) { in affine_grid_generator_backward()
|
| H A D | GridSampler.h | 27 bool align_corners) { in grid_sampler_unnormalize()
43 bool align_corners, scalar_t *grad_in) { in grid_sampler_unnormalize_set_grad()
144 bool align_corners) { in compute_coordinates()
167 bool align_corners) { in grid_sampler_compute_source_index()
182 bool align_corners, in grid_sampler_compute_source_index_set_grad()
222 bool align_corners) { in get_value_bounded()
266 bool align_corners) { in add_value_bounded()
|
| H A D | GridSampler.cpp | 44 bool align_corners) { in grid_sampler_3d_cpu_impl()
209 bool align_corners, std::array<bool,2> output_mask) { in grid_sampler_3d_backward_cpu_impl() argument
451 bool align_corners) { in _grid_sampler_2d_cpu_quantized()
560 bool align_corners) { in _grid_sampler_2d_cpu_fallback()
716 bool align_corners) { in _grid_sampler_2d_cpu_fallback_backward()
921 bool align_corners) { in grid_sampler_2d_cpu()
965 bool align_corners) { in grid_sampler_3d_cpu()
980 int64_t interpolation_mode, int64_t padding_mode, bool align_corners, in grid_sampler_2d_backward_cpu()
1029 int64_t interpolation_mode, int64_t padding_mode, bool align_corners, in grid_sampler_3d_backward_cpu()
1051 bool align_corners in grid_sampler()
|
| H A D | UpSampleBicubic2d.cpp | 117 bool align_corners, in upsample_bicubic2d_backward_out_frame()
187 bool align_corners, in upsample_bicubic2d_backward_kernel()
282 bool align_corners, in upsample_bicubic2d()
293 bool align_corners, in _upsample_bicubic2d_aa()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cuh | 22 scalar_t grid_sampler_unnormalize(scalar_t coord, int size, bool align_corners) { in grid_sampler_unnormalize()
39 bool align_corners, scalar_t *grad_in) { in grid_sampler_unnormalize_set_grad()
152 bool align_corners) { in compute_coordinates()
178 bool align_corners) { in grid_sampler_compute_source_index()
194 bool align_corners, in grid_sampler_compute_source_index_set_grad()
233 bool align_corners) { in get_value_bounded()
285 bool align_corners, in add_value_bounded()
|
| H A D | GridSampler.cpp | 21 bool align_corners) { in grid_sampler_2d_cuda()
33 bool align_corners) { in grid_sampler_3d_cuda()
47 bool align_corners, std::array<bool, 2> output_mask) { in grid_sampler_2d_backward_cuda() argument
66 bool align_corners, std::array<bool,2> output_mask) { in grid_sampler_3d_backward_cuda() argument
|
| H A D | UpSampleBilinear2d.cu | 39 const bool align_corners, in upsample_bilinear2d_out_frame()
87 const bool align_corners, in upsample_bilinear2d_nhwc_out_frame()
141 const bool align_corners, in upsample_bilinear2d_backward_out_frame()
205 const bool align_corners, in upsample_bilinear2d_backward_nhwc_out_frame()
266 bool align_corners, in upsample_bilinear2d_out_cuda_template()
364 bool align_corners, in upsample_bilinear2d_backward_out_cuda_template()
678 bool align_corners, in upsample_gen2d_aa_out_cuda_template()
774 bool align_corners, in upsample_gen2d_aa_backward_out_cuda_template()
|
| H A D | UpSampleLinear1d.cu | 30 const bool align_corners, in upsample_linear1d_out_frame()
77 const bool align_corners, in upsample_linear1d_out_frame_backward()
123 bool align_corners, in upsample_linear1d_out_cuda_template()
166 bool align_corners, in upsample_linear1d_backward_out_cuda_template()
|
| H A D | UpSampleBicubic2d.cu | 28 const bool align_corners, in upsample_bicubic2d_out_frame()
103 const bool align_corners, in upsample_bicubic2d_backward_out_frame()
172 bool align_corners, in upsample_bicubic2d_out_cuda_template()
227 bool align_corners, in upsample_bicubic2d_backward_out_cuda_template()
|
| H A D | GridSampler.cu | 33 bool align_corners) { in grid_sampler_2d_kernel()
163 bool align_corners) { in grid_sampler_3d_kernel()
321 bool align_corners, in grid_sampler_2d_backward_kernel()
527 bool align_corners, in grid_sampler_3d_backward_kernel()
752 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in launch_grid_sampler_2d_forward_kernel()
796 int64_t interpolation_mode, int64_t padding_mode, bool align_corners) { in launch_grid_sampler_3d_forward_kernel()
843 bool align_corners, std::array<bool,2> output_mask) { in launch_grid_sampler_2d_backward_kernel() argument
906 bool align_corners, std::array<bool,2> output_mask) { in launch_grid_sampler_3d_backward_kernel() argument
|
| H A D | UpSampleTrilinear3d.cu | 45 const bool align_corners, in upsample_trilinear3d_out_frame()
129 const bool align_corners, in upsample_trilinear3d_backward_out_frame()
247 bool align_corners, in upsample_trilinear3d_out_cuda_template()
303 bool align_corners, in upsample_trilinear3d_backward_out_cuda_template()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | ScaleFixture.h | 43 bool align_corners, bool mixed_layout, QuantizationInfo output_quantization_info) in setup()
216 bool align_corners) in setup()
237 bool align_corners) in setup()
256 …rpolationPolicy policy, BorderMode border_mode, SamplingPolicy sampling_policy, bool align_corners) in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | image_grad_test_base.py | 98 def resize_nn(t, shape=out_shape, align_corners=align_corners): argument
134 align_corners=False, argument
235 def _gpuVsCpuCase(self, in_shape, out_shape, align_corners, argument
305 def func(input_tensor, align_corners=align_corners): argument
324 def func(input_tensor, align_corners=align_corners): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleKernel.cpp | 567 bool align_corners, in cpu_upsample_linear_channels_last()
1422 bool align_corners, in upsample_generic_Nd_kernel_impl()
1519 bool align_corners, in _separable_upsample_generic_Nd_kernel_impl_single_dim()
1589 bool align_corners, in separable_upsample_generic_Nd_kernel_impl()
1767 bool align_corners, in upsample_linear1d_kernel_impl()
1777 bool align_corners, in upsample_bilinear2d_kernel_impl_float()
1799 bool align_corners, in upsample_bilinear2d_kernel_impl()
1828 bool align_corners, in upsample_bilinear2d_aa_kernel_impl()
1851 bool align_corners, in upsample_trilinear3d_kernel_impl()
1868 bool align_corners, in upsample_bicubic2d_kernel_impl()
[all …]
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Scale.cpp | 41 … SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) in scale_core()
185 …SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners, QuantizationInfo outpu… in scale()
193 …SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners, QuantizationInfo outpu… in scale()
212 …SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners, QuantizationInfo outpu… in scale()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/scale/sve/ |
| H A D | integer.cpp | 43 float sampling_offset, bool align_corners, const Window &window) in u8_sve_scale_nearest()
87 float sampling_offset, bool align_corners, const Window &window) in s16_sve_scale_nearest()
134 bool align_corners, const Window &window) in u8_sve_scale()
149 bool align_corners, const Window &window) in s16_sve_scale()
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/attributes/ |
| H A D | ResizeAttributes.cpp | 77 ResizeAttributes &ResizeAttributes::align_corners(bool align_corners) in align_corners() function in arm_compute::experimental::dynamic_fusion::ResizeAttributes
83 bool ResizeAttributes::align_corners() const in align_corners() function in arm_compute::experimental::dynamic_fusion::ResizeAttributes
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/scale/neon/ |
| H A D | integer.cpp | 37 float sampling_offset, bool align_corners, const Window &window) in u8_neon_scale_nearest()
80 bool align_corners, const Window &window) in u8_neon_scale_bilinear()
316 bool align_corners, const Window &window) in s8_neon_scale_bilinear()
519 float sampling_offset, bool align_corners, const Window &window) in s16_neon_scale_nearest()
562 bool align_corners, const Window &window) in s16_neon_scale_bilinear()
633 bool align_corners, const Window &window) in s8_neon_scale()
647 bool align_corners, const Window &window) in u8_neon_scale()
661 bool align_corners, const Window &window) in s16_neon_scale()
|
| H A D | fp16.cpp | 45 float sampling_offset, bool align_corners, const Window &window) in fp16_neon_scale_nearest()
88 bool align_corners, const Window &window) in fp16_neon_scale_bilinear()
161 bool align_corners, const Window &window) in fp16_neon_scale()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | image_grad.cc | 35 bool align_corners; in ResizeNearestNeighborGradHelper() local
57 bool align_corners; in ResizeBilinearGradHelper() local
75 bool align_corners; in ResizeBicubicGradHelper() local
|
| H A D | image_grad_test.cc | 55 const bool align_corners, const bool half_pixel_centers, in MakeOp()
79 void TestResizedShapeForType(const OpType op_type, const bool align_corners, in TestResizedShapeForType()
112 const bool align_corners, in TestResizeToSmallerAndAlign()
127 const bool align_corners, in TestResizeToLargerAndAlign()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/ |
| H A D | ResizeFixture.h | 58 bool align_corners, QuantizationInfo output_quantization_info) in setup()
227 …Layout data_layout, InterpolationPolicy policy, SamplingPolicy sampling_policy, bool align_corners) in setup()
246 bool align_corners) in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | image_ops_test.py | 412 align_corners=True): argument
551 align_corners=False, argument
666 align_corners=True): argument
827 align_corners=True): argument
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | resize_nearest_neighbor_v1_3.mod.py | 16 def test(name, input0, output_width, output_height, layout, align_corners, argument
213 align_corners = BoolScalar("align_corners", False) variable
|
| D | resize_bilinear_v1_3.mod.py | 21 align_corners, argument
538 align_corners = BoolScalar("align_corners", False) variable
|