Class OpMapperContext

Defined in File op_mapper_registry.h

Nested Relationships

Nested Types

Struct OpMapperContext::FeedInfo

Class Documentation

class cinn::frontend::OpMapperContext

Public Functions

OpMapperContext(const hlir::framework::Scope &scope, const common::Target &target, NetBuilder *builder, std::unordered_map<std::string, Variable> *var_map, std::unordered_map<std::string, std::string> *var_model_to_program_map, std::unordered_set<std::string> *fetch_var_names)

const auto &Scope() const

const auto &Target() const

NetBuilder *Builder() const

void AddVar(const std::string &name, const Variable &var, bool replace = false) const

Variable GetVar(const std::string &name) const

void AddVarModelToProgram(const std::string &name, const std::string &id) const

void AddFetchVarName(const std::string &name) const

void AddFeedInfo(const std::string &name, const FeedInfo &info)

const FeedInfo &GetFeedInfo(const std::string &name) const

struct FeedInfo

Public Members

std::vector<int> shape

common::Type type