Function cinn::lang::CollectTempTensorsFromCtrlDepends

Function Documentation

std::set<ir::Tensor> cinn::lang::CollectTempTensorsFromCtrlDepends(StageMap stages, const std::vector<Tensor> &tensor_args)