Function cinn::lang::detail::CreateCompGraph

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 the extra_depend_stages stored in tensor.stage.

Return

a graph.

Parameters
  • tensors: the input/output tensors of a computation.

  • hide_inline: hide inline tensor nodes.