58 строки
1.5 KiB
C++
58 строки
1.5 KiB
C++
/*!
|
|
* Copyright (c) 2017 by Contributors
|
|
* \file ir_operator.h
|
|
* \brief Common operators of Expr
|
|
*/
|
|
#ifndef TVM_IR_OPERATOR_H_
|
|
#define TVM_IR_OPERATOR_H_
|
|
|
|
#include <algorithm>
|
|
#include "./expr.h"
|
|
#include "./ir.h"
|
|
|
|
namespace tvm {
|
|
|
|
using Halide::likely;
|
|
using Halide::likely_if_innermost;
|
|
// functions
|
|
using Halide::cast;
|
|
using Halide::min;
|
|
using Halide::max;
|
|
using Halide::abs;
|
|
using Halide::select;
|
|
|
|
/*!
|
|
* \brief sum of of source expression over axis
|
|
* \param source The source expression.
|
|
* \param axis List of iteration variables that will be used for reduction.
|
|
*/
|
|
Expr sum(Expr source, Array<IterVar> axis);
|
|
|
|
/*!
|
|
* \brief max of of source expression over axis
|
|
* \param source The source expression.
|
|
* \param axis List of iteration variables that will be used for reduction.
|
|
*/
|
|
Expr max(Expr source, Array<IterVar> axis);
|
|
|
|
/*!
|
|
* \brief max of of source expression over axis
|
|
* \param source The source expression.
|
|
* \param axis List of iteration variables that will be used for reduction.
|
|
*/
|
|
Expr min(Expr source, Array<IterVar> axis);
|
|
|
|
// Unary intrinsic operators
|
|
#define TVM_DECLARE_INTRIN_UNARY(OpName) \
|
|
inline Expr OpName(Expr x) { \
|
|
return ir::Call::make(x.type(), #OpName, {x}, ir::Call::PureExtern); \
|
|
} \
|
|
|
|
TVM_DECLARE_INTRIN_UNARY(exp);
|
|
TVM_DECLARE_INTRIN_UNARY(tanh);
|
|
TVM_DECLARE_INTRIN_UNARY(sigmoid);
|
|
TVM_DECLARE_INTRIN_UNARY(sqrt);
|
|
} // namespace tvm
|
|
|
|
#endif // TVM_IR_OPERATOR_H_
|