19 #ifndef TVM_META_SCHEDULE_SPACE_GENERATOR_H_
20 #define TVM_META_SCHEDULE_SPACE_GENERATOR_H_
22 #include <tvm/ffi/container/array.h>
23 #include <tvm/ffi/function.h>
24 #include <tvm/ffi/reflection/registry.h>
34 namespace meta_schedule {
87 refl::ObjectDef<SpaceGeneratorNode>()
116 static constexpr
const char*
_type_key =
"meta_schedule.SpaceGenerator";
158 Optional<Array<ScheduleRule>> sch_rules, Optional<Array<Postproc>> postprocs,
159 Optional<Map<Mutator, FloatImm>> mutator_probs,
173 Optional<Array<ScheduleRule>> sch_rules,
174 Optional<Array<Postproc>> postprocs,
175 Optional<Map<Mutator, FloatImm>> mutator_probs);
185 Optional<Array<ScheduleRule>> sch_rules,
186 Optional<Array<Postproc>> postprocs,
187 Optional<Map<Mutator, FloatImm>> mutator_probs);
198 Optional<Array<ScheduleRule>> sch_rules,
199 Optional<Array<Postproc>> postprocs,
200 Optional<Map<Mutator, FloatImm>> mutator_probs);
227 static constexpr const
char*
_type_key = "meta_schedule.PySpaceGenerator";
Managed reference class to IRModuleNode.
Definition: module.h:257
Managed reference to TuneContextNode.
Definition: tune_context.h:98
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
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
A managed object in the TVM runtime.
Compilation target object.