Class BaseBuilder

Defined in File base_builder.h

Inheritance Relationships

Derived Types

public cinn::frontend::CinnBuilder (Class CinnBuilder)
public cinn::frontend::NetBuilder (Class NetBuilder)

Class Documentation

class cinn::frontend::BaseBuilder

Subclassed by cinn::frontend::CinnBuilder, cinn::frontend::NetBuilder

Public Functions

BaseBuilder(const std::string &name)

Program Build()

Placeholder CreateInput(const common::Type &type, const std::vector<int> &shape, const std::string &id_hint = "")

Placeholder CreateInput(const Variable &input)

const std::string &name()

const size_t size()

~BaseBuilder()

void AppendInstruction(const Instruction &instr)

Protected Functions

void InferShape(Instruction instr) const

CINN_DISALLOW_COPY_AND_ASSIGN(BaseBuilder)

Protected Attributes

std::string name_

std::vector<Instruction> instrs_

std::vector<Variable> inputs_