read_image_cpu.h 163 Bytes
Newer Older
Francisco Massa's avatar
Francisco Massa committed
1
2
3
4
5
#pragma once

#include "readjpeg_cpu.h"
#include "readpng_cpu.h"

6
7
8
C10_EXPORT torch::Tensor decode_image(
    const torch::Tensor& data,
    int64_t channels = 0);