25 #ifndef NNVM_COMPILER_UTIL_H_ 26 #define NNVM_COMPILER_UTIL_H_ 43 tvm::Array<tvm::Expr> result;
44 for (
auto i : shape) {
45 result.push_back(tvm::make_const(tvm::Int(32), i));
59 return tvm::Array<tvm::Integer>(
ShapeToArray(shape).node_);
63 #endif // NNVM_COMPILER_UTIL_H_
A Shape class that is used to represent shape of each tensor.
Definition: tuple.h:344
tvm::Array< tvm::Expr > ShapeToArray(TShape shape)
Definition: util.h:42
Data structure Tuple and TShape to store dynamic sized shapes.
tvm::Array< tvm::Integer > ShapeToIntArray(TShape shape)
Definition: util.h:58