Searched refs:subgraph_outputs (Results 1 – 1 of 1) sorted by relevance
124 void InitializeGraphOutputs(const std::set<int>& subgraph_outputs) { in InitializeGraphOutputs() argument127 subgraph_outputs_.push_back(subgraph_outputs.count(i) > 0); in InitializeGraphOutputs()366 std::vector<int> subgraph_outputs; member386 op_data->subgraph_outputs.push_back(tensor_index); in Init()481 for (auto tensor_index : op_data->subgraph_outputs) { in Prepare()540 for (auto tensor_index : op_data->subgraph_outputs) { in Eval()