mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
30 lines
579 B
C
30 lines
579 B
C
|
// Copyright (c) OpenMMLab. All rights reserved.
|
||
|
|
|
||
|
|
#ifndef MMDEPLOY_TRANSFORM_UTILS_H
|
||
|
|
#define MMDEPLOY_TRANSFORM_UTILS_H
|
||
|
|
|
||
|
|
#include "core/mat.h"
|
||
|
|
#include "core/tensor.h"
|
||
|
|
|
||
|
|
namespace mmdeploy {
|
||
|
|
/**
|
||
|
|
*
|
||
|
|
* @param src
|
||
|
|
* @param device
|
||
|
|
* @param stream
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
Result<Mat> MakeAvailableOnDevice(const Mat& src, const Device& device, Stream& stream);
|
||
|
|
|
||
|
|
/**
|
||
|
|
*
|
||
|
|
* @param src
|
||
|
|
* @param device
|
||
|
|
* @param stream
|
||
|
|
* @return
|
||
|
|
*/
|
||
|
|
Result<Tensor> MakeAvailableOnDevice(const Tensor& src, const Device& device, Stream& stream);
|
||
|
|
} // namespace mmdeploy
|
||
|
|
|
||
|
|
#endif // MMDEPLOY_TRANSFORM_UTILS_H
|