Function cinn::lang::detail::CreateCompGraphWithInlineTensors

• Defined in File lower_impl.cc

Function Documentation

void cinn::lang::detail::CreateCompGraphWithInlineTensors(common::Graph *graph, const ir::Tensor &t, StageMap stages, std::set<ir::Tensor> *visited)

Add nodes to graph with dependencies. We create a computation graph based on the tensor dependency relations. NOTE The graph will contain the inline tensors so that the dependency will be reserved.

Parameters

• graph: The graph

• t: The tensor.

• stages: The stage map.