Home
last modified time | relevance | path

Searched refs:LiveTensor (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.cc121 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 …]
Dgraph_memory.h31 struct LiveTensor { struct
40 std::vector<LiveTensor> live_tensors; argument