Searched refs:graph_output_tensors (Results 1 – 3 of 3) sorted by relevance
88 std::vector<tensor::TensorPtr> graph_output_tensors; member
1372 graph_output_info->graph_output_tensors.emplace_back(output_tensor); in HandleOpOutputs()2305 graph_output_info.graph_output_tensors.clear(); in RunOpsInGraphImpl()2311 AddGradAddrToBucket(graph_id, graph_output_info.graph_output_tensors); in RunOpsInGraphImpl()
763 graph_output_info.graph_output_tensors.clear(); in RunGraphBySingleOp()768 … graph_compiler_->AddGradAddrToBucket(graph->graph_id(), graph_output_info.graph_output_tensors); in RunGraphBySingleOp()