Searched refs:LiveTensor (Results 1 – 2 of 2) sorted by relevance
122 static GraphMemory::LiveTensor* FindOrCreateLiveTensor( in FindOrCreateLiveTensor()124 std::unordered_map<string, GraphMemory::LiveTensor*>* live_tensors, in FindOrCreateLiveTensor()125 std::deque<GraphMemory::LiveTensor>* device_tensors) { in FindOrCreateLiveTensor()127 GraphMemory::LiveTensor* live; in FindOrCreateLiveTensor()130 GraphMemory::LiveTensor temp; in FindOrCreateLiveTensor()147 const GraphMemory::LiveTensor* _tensor) in Event()152 const GraphMemory::LiveTensor* tensor;168 std::unordered_map<string, LiveTensor*> live_tensors; in InferFromTrace()169 std::unordered_map<string, std::deque<LiveTensor>> live_tensors_per_device; in InferFromTrace()177 std::deque<LiveTensor>& device_tensors = in InferFromTrace()[all …]
31 struct LiveTensor { struct40 std::vector<LiveTensor> live_tensors; argument