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>()
131 TVM_FFI_ICHECK(data !=
nullptr);
165 ffi::Optional<ffi::Array<ScheduleRule>> sch_rules,
166 ffi::Optional<ffi::Array<Postproc>> postprocs,
167 ffi::Optional<ffi::Map<Mutator, FloatImm>> mutator_probs,
181 ffi::Function schedule_fn, ffi::Optional<ffi::Array<ScheduleRule>> sch_rules,
182 ffi::Optional<ffi::Array<Postproc>> postprocs,
183 ffi::Optional<ffi::Map<Mutator, FloatImm>> mutator_probs);
193 ffi::Array<SpaceGenerator, void> space_generators,
194 ffi::Optional<ffi::Array<ScheduleRule>> sch_rules,
195 ffi::Optional<ffi::Array<Postproc>> postprocs,
196 ffi::Optional<ffi::Map<Mutator, FloatImm>> mutator_probs);
207 ffi::Function f_block_filter, ffi::Optional<ffi::Array<ScheduleRule>> sch_rules,
208 ffi::Optional<ffi::Array<Postproc>> postprocs,
209 ffi::Optional<ffi::Map<Mutator, FloatImm>> mutator_probs);
Managed reference class to IRModuleNode.
Definition: module.h:256
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:308
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
A managed object in the TVM runtime.
Compilation target object.