19 #ifndef TVM_META_SCHEDULE_ARG_INFO_H_
20 #define TVM_META_SCHEDULE_ARG_INFO_H_
22 #include <tvm/ffi/container/shape.h>
23 #include <tvm/ffi/reflection/registry.h>
31 namespace meta_schedule {
36 static constexpr
const char*
_type_key =
"meta_schedule.ArgInfo";
43 virtual ObjectRef
AsJSON()
const = 0;
50 class ArgInfo :
public runtime::ObjectRef {
88 refl::ObjectDef<TensorInfoNode>()
93 static constexpr
const char*
_type_key =
"meta_schedule.TensorInfo";
Managed reference class to IRModuleNode.
Definition: module.h:257
Runtime primitive data type.
Definition: data_type.h:47
Managed reference to PrimFuncNode.
Definition: function.h:131
IRModule that holds the functions and type definitions.
Definition: repr_printer.h:91
tvm::PrimExpr mod(const tvm::PrimExpr &a, const tvm::PrimExpr &b)
Definition: broadcast.h:306
Tensor shape(const Tensor &src, DataType dtype, const std::string name="T_shape", const std::string tag=kInjective)
Get the shape of input tensor.
Definition: transform.h:1945
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Definitions and helper macros for IR/AST nodes.
A managed object in the TVM runtime.