Searched refs:GetNodeOutputTensor (Results 1 – 2 of 2) sorted by relevance
285 tensor::TensorPtr GetNodeOutputTensor(const session::KernelWithIndex &output_index) const { in GetNodeOutputTensor() function
1350 auto tensor = graph.GetNodeOutputTensor(std::make_pair(kernel, j)); in SyncNodeOutputTensors()