mirror of https://github.com/alibaba/MNN.git
29 lines
667 B
C++
29 lines
667 B
C++
|
|
#ifndef CPURESIZECACHE_HPP
|
|
#define CPURESIZECACHE_HPP
|
|
#include <map>
|
|
#include <MNN/Tensor.hpp>
|
|
#include "MNN_generated.h"
|
|
|
|
namespace MNN {
|
|
// FIXME: Move outside
|
|
class MNN_PUBLIC CPUResizeCache {
|
|
public:
|
|
CPUResizeCache() {
|
|
// Do nothing
|
|
}
|
|
~ CPUResizeCache() {
|
|
// Do nothing
|
|
}
|
|
Tensor* findCacheTensor(const Tensor* src, MNN_DATA_FORMAT format) const;
|
|
// Return cache tensor
|
|
void pushCacheTensor(std::shared_ptr<Tensor> dst, const Tensor* src, MNN_DATA_FORMAT format);
|
|
void reset();
|
|
void release();
|
|
private:
|
|
std::map<std::pair<const Tensor*, MNN_DATA_FORMAT>, std::shared_ptr<Tensor>> mFormatCache;
|
|
};
|
|
}
|
|
|
|
#endif
|