Searched refs:device_tensors (Results 1 – 1 of 1) sorted by relevance
125 std::deque<GraphMemory::LiveTensor>* device_tensors) { in FindOrCreateLiveTensor() argument135 device_tensors->push_front(temp); in FindOrCreateLiveTensor()136 live = &device_tensors->front(); in FindOrCreateLiveTensor()177 std::deque<LiveTensor>& device_tensors = in InferFromTrace() local184 node_stats.node_name(), i, &live_tensors, &device_tensors); in InferFromTrace()