decode_jpeg.h 359 B

1234567891011121314151617
  1. #pragma once
  2. #include <torch/types.h>
  3. #include "../image_read_mode.h"
  4. namespace vision {
  5. namespace image {
  6. C10_EXPORT torch::Tensor decode_jpeg(
  7. const torch::Tensor& data,
  8. ImageReadMode mode = IMAGE_READ_MODE_UNCHANGED);
  9. C10_EXPORT int64_t _jpeg_version();
  10. C10_EXPORT bool _is_compiled_against_turbo();
  11. } // namespace image
  12. } // namespace vision