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.cc463 AllocStats& dev_allocs_stats = allocs_map[dev_allocator]; in ReportAllocsOnResourceExhausted() local
472 dev_allocs_stats.total_bytes += cur_bytes; in ReportAllocsOnResourceExhausted()
473 dev_allocs_stats.total_nodes++; in ReportAllocsOnResourceExhausted()
474 dev_allocs_stats.nodes_by_size[cur_bytes].push_back( in ReportAllocsOnResourceExhausted()
483 const AllocStats& dev_allocs_stats = dev_allocs_it.second; in ReportAllocsOnResourceExhausted() local
490 for (auto it = dev_allocs_stats.nodes_by_size.rbegin(); in ReportAllocsOnResourceExhausted()
491 it != dev_allocs_stats.nodes_by_size.rend(); ++it) { in ReportAllocsOnResourceExhausted()
499 dev_allocs_stats.total_bytes * kMaxAllocReportFraction) { in ReportAllocsOnResourceExhausted()
506 int64 remain_nodes = dev_allocs_stats.total_nodes - reported_nodes; in ReportAllocsOnResourceExhausted()
507 int64 remain_bytes = dev_allocs_stats.total_bytes - reported_bytes; in ReportAllocsOnResourceExhausted()