Searched refs:live_tensors (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_memory.cc | 123 std::unordered_map<string, GraphMemory::LiveTensor*>* live_tensors, in FindOrCreateLiveTensor() argument
127 auto it = live_tensors->find(name); in FindOrCreateLiveTensor()
128 if (it == live_tensors->end()) { in FindOrCreateLiveTensor()
136 (*live_tensors)[name] = live; in FindOrCreateLiveTensor()
167 std::unordered_map<string, LiveTensor*> live_tensors; in InferFromTrace() local
183 node_stats.node_name(), i, &live_tensors, &device_tensors); in InferFromTrace()
233 input_node, position, &live_tensors, in InferFromTrace()
284 peak_mem_usage.live_tensors.clear(); in InferFromTrace()
285 peak_mem_usage.live_tensors.reserve(live_at_peak.size()); in InferFromTrace()
287 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 | 77 live_tensors = snapshot[1]
78 for tensor in live_tensors:
|
| D | cluster_test.py | 85 live_tensors = snapshot[1]
86 self.assertEqual(5, len(live_tensors))
|
| D | cluster_wrapper.cc | 322 for (size_t i = 0; i < usage.live_tensors.size(); ++i) { in PYBIND11_MODULE()
323 const auto& live_tensor = usage.live_tensors[i]; in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 567 for (const auto& live : mem_usage.live_tensors) { in SchedulingPass()
1041 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates()
1050 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates()
|