tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::meta_schedule::BuilderInputNode Class Reference

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

#include <builder.h>

Inheritance diagram for tvm::meta_schedule::BuilderInputNode:
Collaboration diagram for tvm::meta_schedule::BuilderInputNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (BuilderInputNode, runtime::Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

IRModule mod
 The IRModule to be built. More...
 
Target target
 The target to be built for. More...
 
Optional< Map< String, runtime::NDArray > > params
 Parameters for Relax build module. More...
 

Static Public Attributes

static constexpr const char * _type_key = "meta_schedule.BuilderInput"
 

Detailed Description

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

Member Function Documentation

◆ RegisterReflection()

static void tvm::meta_schedule::BuilderInputNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::BuilderInputNode::TVM_DECLARE_FINAL_OBJECT_INFO ( BuilderInputNode  ,
runtime::Object   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::meta_schedule::BuilderInputNode::_type_key = "meta_schedule.BuilderInput"
staticconstexpr

◆ mod

IRModule tvm::meta_schedule::BuilderInputNode::mod

The IRModule to be built.

◆ params

Optional<Map<String, runtime::NDArray> > tvm::meta_schedule::BuilderInputNode::params

Parameters for Relax build module.

◆ target

Target tvm::meta_schedule::BuilderInputNode::target

The target to be built for.


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