Function cinn::lang::detail::CreateCompGraph
Defined in File lower_impl.cc
Function Documentation
-
std::unique_ptr<common::Graph>
cinn::lang::detail
::
CreateCompGraph
(const std::vector<ir::Tensor> &tensors, StageMap stages, bool hide_inline = false) Create a computation graph using a tensor set. It will deduce the temporary tensors not in the
tensors
. It consider theextra_depend_stages
stored in tensor.stage.- Return
a graph.
- Parameters
tensors
: the input/output tensors of a computation.hide_inline
: hide inline tensor nodes.