Searched refs:live_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 414 const auto& live_tensor = usage.live_tensors[i]; in TF_DeterminePeakMemoryUsage() local
416 PyTuple_SetItem(live, 0, PyString_FromString(live_tensor.node.c_str())); in TF_DeterminePeakMemoryUsage()
417 PyTuple_SetItem(live, 1, PyInt_FromLong(live_tensor.output_id)); in TF_DeterminePeakMemoryUsage()
418 PyTuple_SetItem(live, 2, PyLong_FromLong(live_tensor.memory_used)); in TF_DeterminePeakMemoryUsage()
419 PyTuple_SetItem(live, 3, PyLong_FromLong(live_tensor.allocation_time.count())); in TF_DeterminePeakMemoryUsage()
420 PyTuple_SetItem(live, 4, PyLong_FromLong(live_tensor.deallocation_time.count())); in TF_DeterminePeakMemoryUsage()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 1006 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates() local
1007 if (live_tensor.allocation_time > peak_time) { in IdentifySwappingCandidates()
1008 peak_time = live_tensor.allocation_time; in IdentifySwappingCandidates()
1015 for (const auto& live_tensor : mem_usage.live_tensors) { in IdentifySwappingCandidates() local
1016 if (live_tensor.memory_used <= 1024) { in IdentifySwappingCandidates()
1020 if (live_tensor.deallocation_time - live_tensor.allocation_time <= in IdentifySwappingCandidates()
1023 VLOG(1) << "Not enough time to swap: skipping " << live_tensor.node; in IdentifySwappingCandidates()
1027 if (skip_list->find(live_tensor.node) != skip_list->end()) { in IdentifySwappingCandidates()
1031 graph.GetOutputPort(live_tensor.node, live_tensor.output_id); in IdentifySwappingCandidates()
1037 mem_info.memory_used = live_tensor.memory_used; in IdentifySwappingCandidates()
[all …]
|