Searched refs:live_tensors (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_memory.cc | 124 std::unordered_map<string, GraphMemory::LiveTensor*>* live_tensors, in FindOrCreateLiveTensor() argument
128 auto it = live_tensors->find(name); in FindOrCreateLiveTensor()
129 if (it == live_tensors->end()) { in FindOrCreateLiveTensor()
137 (*live_tensors)[name] = live; in FindOrCreateLiveTensor()
168 std::unordered_map<string, LiveTensor*> live_tensors; in InferFromTrace() local
184 node_stats.node_name(), i, &live_tensors, &device_tensors); in InferFromTrace()
234 input_node, position, &live_tensors, in InferFromTrace()
285 peak_mem_usage.live_tensors.clear(); in InferFromTrace()
286 peak_mem_usage.live_tensors.reserve(live_at_peak.size()); in InferFromTrace()
288 peak_mem_usage.live_tensors.push_back(*live); in InferFromTrace()
|
| D | graph_memory_test.cc | 59 for (const auto& t : mem_usage.live_tensors) { in TEST_F()
89 for (const auto& t : mem_usage.live_tensors) { in TEST_F()
113 for (const auto& t : cpu_mem.live_tensors) { in TEST_F()
126 for (const auto& t : gpu_mem.live_tensors) { in TEST_F()
153 for (const auto& t : gpu_mem.live_tensors) { in TEST_F()
180 for (const auto& t : new_gpu_mem.live_tensors) { in TEST_F()
216 for (const auto& t : mem.live_tensors) { in TEST_F()
|
| D | graph_memory.h | 40 std::vector<LiveTensor> live_tensors; member
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cost_analyzer.py | 80 live_tensors = snapshot[1]
81 for tensor in live_tensors:
|
| D | cluster_test.py | 85 live_tensors = snapshot[1]
86 self.assertEqual(15, len(live_tensors))
|
| D | cluster.i | 412 PyObject* per_device = PyList_New(usage.live_tensors.size()); in TF_DeterminePeakMemoryUsage()
413 for (int i = 0; i < usage.live_tensors.size(); ++i) { in TF_DeterminePeakMemoryUsage()
414 const auto& live_tensor = usage.live_tensors[i]; in TF_DeterminePeakMemoryUsage()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 558 for (const auto& live : mem_usage.live_tensors) { in SchedulingPass()
1006 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates()
1015 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates()
|