30 lines
1.2 KiB
C++
30 lines
1.2 KiB
C++
#include "image.h"
|
|
|
|
#include <ATen/core/op_registration/op_registration.h>
|
|
|
|
namespace vision {
|
|
namespace image {
|
|
|
|
static auto registry =
|
|
torch::RegisterOperators()
|
|
.op("image::decode_gif", &decode_gif)
|
|
.op("image::decode_png(Tensor data, int mode, bool apply_exif_orientation=False) -> Tensor",
|
|
&decode_png)
|
|
.op("image::encode_png", &encode_png)
|
|
.op("image::decode_jpeg(Tensor data, int mode, bool apply_exif_orientation=False) -> Tensor",
|
|
&decode_jpeg)
|
|
.op("image::decode_webp(Tensor encoded_data, int mode) -> Tensor",
|
|
&decode_webp)
|
|
.op("image::encode_jpeg", &encode_jpeg)
|
|
.op("image::read_file", &read_file)
|
|
.op("image::write_file", &write_file)
|
|
.op("image::decode_image(Tensor data, int mode, bool apply_exif_orientation=False) -> Tensor",
|
|
&decode_image)
|
|
.op("image::decode_jpegs_cuda", &decode_jpegs_cuda)
|
|
.op("image::encode_jpegs_cuda", &encode_jpegs_cuda)
|
|
.op("image::_jpeg_version", &_jpeg_version)
|
|
.op("image::_is_compiled_against_turbo", &_is_compiled_against_turbo);
|
|
|
|
} // namespace image
|
|
} // namespace vision
|