MNN/source/backend/cpu/CPUTensorConvert.hpp

//
//  CPUTensorConvert.hpp
//  MNN
//
//  Created by MNN on 2018/08/04.
//  Copyright © 2018, Alibaba Group Holding Limited
//

#ifndef CPUTensorConvert_hpp
#define CPUTensorConvert_hpp

#include "core/Execution.hpp"
#include "Tensor_generated.h"
#include "compute/CommonOptFunction.h"
namespace MNN {

class CPUTensorConverter {
public:
    static std::tuple<int, int, int> splitDimensions(const halide_buffer_t& ib, MNN_DATA_FORMAT source);
    static ErrorCode convert(const Tensor* input, const Tensor* output, const CoreFunctions* core = nullptr, int tId = 0, int numberThread = 1);
    static ErrorCode convert(const void* inputRaw, void* outputRaw, MNN_DATA_FORMAT inputFormat, MNN_DATA_FORMAT outputFormat, int batch, int area, int channel, int bytes, const CoreFunctions* core, int tId = 0, int numberThread = 1);
};

} // namespace MNN

#endif /* CPUTensorConvert_hpp */