Searched refs:LiveTensor (Results 1 – 2 of 2) sorted by relevance
121 static GraphMemory::LiveTensor* FindOrCreateLiveTensor( in FindOrCreateLiveTensor()123 std::unordered_map<string, GraphMemory::LiveTensor*>* live_tensors, in FindOrCreateLiveTensor()124 std::deque<GraphMemory::LiveTensor>* device_tensors) { in FindOrCreateLiveTensor()126 GraphMemory::LiveTensor* live; in FindOrCreateLiveTensor()129 GraphMemory::LiveTensor temp; in FindOrCreateLiveTensor()146 const GraphMemory::LiveTensor* _tensor) in Event()151 const GraphMemory::LiveTensor* tensor;167 std::unordered_map<string, LiveTensor*> live_tensors; in InferFromTrace()168 std::unordered_map<string, std::deque<LiveTensor>> live_tensors_per_device; in InferFromTrace()176 std::deque<LiveTensor>& device_tensors = in InferFromTrace()[all …]
31 struct LiveTensor { struct40 std::vector<LiveTensor> live_tensors; argument