encode_jpeg.h 213 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
#pragma once

#include <torch/types.h>

namespace vision {
namespace image {

C10_EXPORT torch::Tensor encode_jpeg(
    const torch::Tensor& data,
    int64_t quality);

} // namespace image
} // namespace vision