mirror of https://github.com/alibaba/MNN.git
14 lines
436 B
C++
14 lines
436 B
C++
#ifndef OpCommonUtils_hpp
|
|
#define OpCommonUtils_hpp
|
|
#include <MNN/Tensor.hpp>
|
|
namespace MNN {
|
|
struct Op;
|
|
class MNN_PUBLIC OpCommonUtils {
|
|
public:
|
|
static void broastCastComputeDim(int* dims, int* stride, int* iStride0, int* iStride1, const Tensor* input0, const Tensor* input1, const Tensor* output);
|
|
static std::vector<std::tuple<int, int, int>> computeReduceDims(const std::vector<Tensor*>& inputs, const Op* op);
|
|
};
|
|
}
|
|
|
|
#endif
|