Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.cc225 SE_AllocatorStats c_stats; in GetAllocatorStats() local
227 &c_stats)) { in GetAllocatorStats()
229 stats.num_allocs = c_stats.num_allocs; in GetAllocatorStats()
230 stats.bytes_in_use = c_stats.bytes_in_use; in GetAllocatorStats()
231 stats.peak_bytes_in_use = c_stats.peak_bytes_in_use; in GetAllocatorStats()
232 stats.largest_alloc_size = c_stats.largest_alloc_size; in GetAllocatorStats()
233 if (c_stats.has_bytes_limit) { in GetAllocatorStats()
234 stats.bytes_limit = c_stats.bytes_limit; in GetAllocatorStats()
236 stats.bytes_reserved = c_stats.bytes_reserved; in GetAllocatorStats()
237 stats.peak_bytes_reserved = c_stats.peak_bytes_reserved; in GetAllocatorStats()
[all …]
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc305 SP_AllocatorStats c_stats{SP_ALLOCATORSTATS_STRUCT_SIZE}; in GetAllocatorStats() local
307 stream_executor_->get_allocator_stats(&device_, &c_stats); in GetAllocatorStats()
311 port::Status status = ValidateSPAllocatorStats(c_stats); in GetAllocatorStats()
317 stats.num_allocs = c_stats.num_allocs; in GetAllocatorStats()
318 stats.bytes_in_use = c_stats.bytes_in_use; in GetAllocatorStats()
319 stats.peak_bytes_in_use = c_stats.peak_bytes_in_use; in GetAllocatorStats()
320 stats.largest_alloc_size = c_stats.largest_alloc_size; in GetAllocatorStats()
321 if (c_stats.has_bytes_limit) { in GetAllocatorStats()
322 stats.bytes_limit = c_stats.bytes_limit; in GetAllocatorStats()
324 stats.bytes_reserved = c_stats.bytes_reserved; in GetAllocatorStats()
[all …]