Struct ComputationContext

Defined in File computation.cc

Struct Documentation

struct cinn::frontend::ComputationContext

Public Members

Target target

void *stream

std::shared_ptr<hlir::framework::Graph> graph

std::shared_ptr<hlir::framework::Scope> scope

std::shared_ptr<hlir::framework::Program> program

std::shared_ptr<hlir::framework::GraphCompiler> graph_compiler

CinnComputation::CompileOptions compile_options

std::vector<hlir::framework::Tensor> inputs

std::vector<hlir::framework::Tensor> outputs

std::unordered_map<std::string, Variable> varmap

std::unordered_map<std::string, std::string> varmap_paddle2program