interpolate_aa.cpp 2.6 KB
Newer Older
1
2
3
4
5
6
7
#include "interpolate_aa.h"

#include <torch/types.h>

namespace vision {
namespace ops {

8
at::Tensor _interpolate_bilinear2d_aa(
9
10
11
12
13
14
    const at::Tensor& input, // Input image
    at::IntArrayRef output_size, // Output image size
    bool align_corners) // The flag to align corners
{
  static auto op =
      c10::Dispatcher::singleton()
15
16
          .findSchemaOrThrow("torchvision::_interpolate_bilinear2d_aa", "")
          .typed<decltype(_interpolate_bilinear2d_aa)>();
17
18
19
  return op.call(input, output_size, align_corners);
}

20
at::Tensor _interpolate_bicubic_aa(
21
22
23
24
25
26
    const at::Tensor& input, // Input image
    at::IntArrayRef output_size, // Output image size
    bool align_corners) // The flag to align corners
{
  static auto op =
      c10::Dispatcher::singleton()
27
28
          .findSchemaOrThrow("torchvision::_interpolate_bicubic2d_aa", "")
          .typed<decltype(_interpolate_bicubic2d_aa)>();
29
30
31
  return op.call(input, output_size, align_corners);
}

32
33
namespace detail {

34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
at::Tensor _interpolate_bilinear2d_aa_backward(
    const at::Tensor& grad_output,
    at::IntArrayRef output_size,
    at::IntArrayRef input_size,
    bool align_corners) {
  static auto op =
      c10::Dispatcher::singleton()
          .findSchemaOrThrow(
              "torchvision::_interpolate_bilinear2d_aa_backward", "")
          .typed<decltype(_interpolate_bilinear2d_aa_backward)>();
  return op.call(grad_output, output_size, output_size, align_corners);
}

at::Tensor _interpolate_bicubic2d_aa_backward(
    const at::Tensor& grad_output,
    at::IntArrayRef output_size,
    at::IntArrayRef input_size,
    bool align_corners) {
  static auto op =
      c10::Dispatcher::singleton()
          .findSchemaOrThrow(
              "torchvision::_interpolate_bicubic2d_aa_backward", "")
          .typed<decltype(_interpolate_bicubic2d_aa_backward)>();
  return op.call(grad_output, output_size, output_size, align_corners);
}
59
60
61
62
63

} // namespace detail

TORCH_LIBRARY_FRAGMENT(torchvision, m) {
  m.def(TORCH_SELECTIVE_SCHEMA(
64
65
66
67
68
      "torchvision::_interpolate_bilinear2d_aa(Tensor input, int[] output_size, bool align_corners) -> Tensor"));
  m.def(TORCH_SELECTIVE_SCHEMA(
      "torchvision::_interpolate_bicubic2d_aa(Tensor input, int[] output_size, bool align_corners) -> Tensor"));
  m.def(TORCH_SELECTIVE_SCHEMA(
      "torchvision::_interpolate_bilinear2d_aa_backward(Tensor input, int[] output_size, int[] input_size, bool align_corners) -> Tensor"));
69
  m.def(TORCH_SELECTIVE_SCHEMA(
70
      "torchvision::_interpolate_bicubic2d_aa_backward(Tensor input, int[] output_size, int[] input_size, bool align_corners) -> Tensor"));
71
72
73
74
}

} // namespace ops
} // namespace vision