tvm::s_tir::meta_schedule::BuilderInputObj Class Reference

The builder's input, containing an IRModule and the target. More...

#include <builder.h>

Inheritance diagram for tvm::s_tir::meta_schedule::BuilderInputObj:

Collaboration diagram for tvm::s_tir::meta_schedule::BuilderInputObj:

Public Member Functions
	BuilderInputObj (ffi::UnsafeInit tag)
	BuilderInputObj (Target target)
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.meta_schedule.BuilderInput", BuilderInputObj, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
IRModule	mod
	The IRModule to be built.
Target	target
	The target to be built for.
ffi::Optional< ffi::Map< ffi::String, ffi::Tensor > >	params
	Parameters for Relax build module.

Detailed Description

The builder's input, containing an IRModule and the target.

Constructor & Destructor Documentation

BuilderInputObj() [1/2]

tvm::s_tir::meta_schedule::BuilderInputObj::BuilderInputObj ( ffi::UnsafeInit  tag )

inlineexplicit

BuilderInputObj() [2/2]

tvm::s_tir::meta_schedule::BuilderInputObj::BuilderInputObj ( Target  target )

inlineexplicit

Member Function Documentation

RegisterReflection()

static void tvm::s_tir::meta_schedule::BuilderInputObj::RegisterReflection ( )

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::s_tir::meta_schedule::BuilderInputObj::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"s_tir.meta_schedule.BuilderInput"	,
		BuilderInputObj	,
		ffi::Object
	)

Member Data Documentation

mod

IRModule tvm::s_tir::meta_schedule::BuilderInputObj::mod

The IRModule to be built.

params

ffi::Optional<ffi::Map<ffi::String, ffi::Tensor> > tvm::s_tir::meta_schedule::BuilderInputObj::params

Parameters for Relax build module.

target

Target tvm::s_tir::meta_schedule::BuilderInputObj::target

The target to be built for.

---

The documentation for this class was generated from the following file:

• include/tvm/s_tir/meta_schedule/builder.h