Class PaddleModelToProgram

• Defined in File paddle_model_to_program.h

Class Documentation

class cinn::frontend::PaddleModelToProgram

Unnamed Group

void AddOpMapper_feed()

void AddOpMapper_fetch()

void AddOpMapper_scale()

void AddOpMapper_mul()

void AddOpMapper_relu()

void AddOpMapper_elementwise_add()

void AddOpMapper_elementwise_mul()

void AddOpMapper_elementwise_div()

void AddOpMapper_elementwise_sub()

void AddOpMapper_conv2d()

void AddOpMapper_batchnorm()

void AddOpMapper_pool2d()

void AddOpMapper_softmax()

void AddOpMapper_relu6()

void AddOpMapper_depthwise_conv2d()

void AddOpMapper_sigmoid()

void AddOpMapper_slice()

void AddOpMapper_dropout_infer()

void AddOpMapper_matmul()

void AddOpMapper_reshape2()

void AddOpMapper_concat()

void AddOpMapper_assign()

void AddOpMapper_fill_constant()

void AddOpMapper_transpose2()

void AddOpMapper_exp()

Public Functions

PaddleModelToProgram(hlir::framework::Scope *scope, const common::Target &target)

std::unique_ptr<Program> operator()(const std::string &model_dir, bool is_combined)

void AddOp(const paddle::cpp::OpDesc &op_desc)

const absl::flat_hash_map<std::string, Variable> &var_map() const

const absl::flat_hash_map<std::string, std::string> &var_model_to_program_map()

const absl::flat_hash_set<std::string> &fetch_names()

Protected Functions

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

Variable GetVar(const std::string &name)

void TransposeVar(const std::string &name)

void ReverseHWVar(const std::string &name)