Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc461 AllocStats& dev_allocs_stats = allocs_map[dev_allocator]; in ReportAllocsOnResourceExhausted() local
470 dev_allocs_stats.total_bytes += cur_bytes; in ReportAllocsOnResourceExhausted()
471 dev_allocs_stats.total_nodes++; in ReportAllocsOnResourceExhausted()
472 dev_allocs_stats.nodes_by_size[cur_bytes].push_back( in ReportAllocsOnResourceExhausted()
481 const AllocStats& dev_allocs_stats = dev_allocs_it.second; in ReportAllocsOnResourceExhausted() local
488 for (auto it = dev_allocs_stats.nodes_by_size.rbegin(); in ReportAllocsOnResourceExhausted()
489 it != dev_allocs_stats.nodes_by_size.rend(); ++it) { in ReportAllocsOnResourceExhausted()
497 dev_allocs_stats.total_bytes * kMaxAllocReportFraction) { in ReportAllocsOnResourceExhausted()
504 int64 remain_nodes = dev_allocs_stats.total_nodes - reported_nodes; in ReportAllocsOnResourceExhausted()
505 int64 remain_bytes = dev_allocs_stats.total_bytes - reported_bytes; in ReportAllocsOnResourceExhausted()