Function cinn::lang::GetTempBuffers

Defined in File lower.cc

Function Documentation

std::vector<ir::Buffer> cinn::lang::GetTempBuffers(const std::vector<Tensor> &tensor_args, const poly::StageMap &stage_map, Expr body): Collect the temporary tensors from a computational graph.