Home
last modified time | relevance | path

Searched defs:align_corners (Results 1 – 25 of 99) sorted by relevance

1234

/external/pytorch/aten/src/ATen/native/
DAffineGridGenerator.cpp19 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()
DGridSampler.h27 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()
DUpSampleBicubic2d.cpp117 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()
DUpSampleBilinear2d.cpp163 bool align_corners, in upsample_bilinear2d()
174 bool align_corners, in _upsample_bilinear2d_aa()
/external/ComputeLibrary/src/dynamic_fusion/sketch/attributes/
DResizeAttributes.cpp77 ResizeAttributes &ResizeAttributes::align_corners(bool align_corners) in align_corners() argument
83 bool ResizeAttributes::align_corners() const in align_corners() function in arm_compute::experimental::dynamic_fusion::ResizeAttributes
/external/pytorch/aten/src/ATen/native/cuda/
DGridSampler.cpp21 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
DGridSampler.cuh22 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()
DUpSampleLinear1d.cu30 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()
DUpSampleBicubic2d.cu28 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()
/external/ComputeLibrary/tests/validation/fixtures/
DScaleFixture.h43 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()
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad.cc35 bool align_corners; in ResizeNearestNeighborGradHelper() local
57 bool align_corners; in ResizeBilinearGradHelper() local
75 bool align_corners; in ResizeBicubicGradHelper() local
Dimage_grad_test.cc55 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()
/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/
DResizeFixture.h58 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()
/external/ComputeLibrary/src/cpu/kernels/scale/sve/
Dinteger.cpp43 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()
Dqasymm8_signed.cpp44 … float sampling_offset, bool align_corners, const Window &window) in qasymm8_signed_sve_scale_nearest()
91 bool align_corners, const Window &window) in qasymm8_signed_sve_scale()
Dfp16.cpp45 float sampling_offset, bool align_corners, const Window &window) in fp16_sve_scale_nearest()
92 bool align_corners, const Window &window) in fp16_sve_scale()
Dfp32.cpp44 float sampling_offset, bool align_corners, const Window &window) in fp32_sve_scale_nearest()
91 bool align_corners, const Window &window) in fp32_sve_scale()
Dqasymm8.cpp44 float sampling_offset, bool align_corners, const Window &window) in qasymm8_sve_scale_nearest()
91 bool align_corners, const Window &window) in qasymm8_sve_scale()
/external/ComputeLibrary/tests/validation/reference/
DScale.cpp41 … 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()
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py98 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
Dimage_grad_d9m_test.py75 def testExceptionThrowing(self, align_corners, half_pixel_centers, data_type): argument
129 def testDeterministicGradients(self, align_corners, half_pixel_centers, argument
/external/ComputeLibrary/src/cpu/kernels/scale/neon/
Dfp16.cpp45 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()
/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dvision.h62 std::optional<bool> align_corners) { in grid_sample()
/external/executorch/backends/xnnpack/partition/graphs/
Dbilinear_2d.py19 def __init__(self, align_corners): argument
/external/ComputeLibrary/src/core/utils/
DScaleUtils.cpp31 …ute::scale_utils::calculate_resize_ratio(size_t input_size, size_t output_size, bool align_corners) in calculate_resize_ratio()

1234