Searched refs:memory_stats (Results 1 – 14 of 14) sorted by relevance
26 from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import BytesInUse27 from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import BytesLimit28 from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import MaxBytesInUse
139 step_stat.memory_stats().temp_memory_size()); in AddMemoryStats()142 step_stat.memory_stats().persistent_memory_size()); in AddMemoryStats()146 step_stat.memory_stats().temp_memory_size()); in AddMemoryStats()149 step_stat.memory_stats().persistent_memory_size()); in AddMemoryStats()
21 from tensorflow.contrib.memory_stats.ops import gen_memory_stats_ops
1417 PartitionBucketMemoryStats memory_stats[kMaxReportableBuckets]; in PartitionDumpStats() local1422 PartitionDumpBucketStats(&memory_stats[i], &partition->buckets()[i]); in PartitionDumpStats()1431 if (memory_stats[i].is_valid) { in PartitionDumpStats()1432 stats.total_resident_bytes += memory_stats[i].resident_bytes; in PartitionDumpStats()1433 stats.total_active_bytes += memory_stats[i].active_bytes; in PartitionDumpStats()1434 stats.total_decommittable_bytes += memory_stats[i].decommittable_bytes; in PartitionDumpStats()1435 stats.total_discardable_bytes += memory_stats[i].discardable_bytes; in PartitionDumpStats()1437 dumper->PartitionsDumpBucketStats(partition_name, &memory_stats[i]); in PartitionDumpStats()
321 const MemoryStats& memory_stats) { in RecordMemoryStats() argument324 max_mem_usage_[id].temp_memory_size = memory_stats.temp_memory_size(); in RecordMemoryStats()326 memory_stats.persistent_memory_size(); in RecordMemoryStats()327 for (int64 alloc_id : memory_stats.persistent_tensor_alloc_ids()) { in RecordMemoryStats()
140 void RecordMemoryStats(const Node* node, const MemoryStats& memory_stats);
104 GENERATE_CONTRIB_OP_LIBRARY(memory_stats "${tensorflow_source_dir}/tensorflow/contrib/memory_stats/…
298 tensorflow/contrib/memory_stats299 tensorflow/contrib/memory_stats/kernels300 tensorflow/contrib/memory_stats/ops301 tensorflow/contrib/memory_stats/python302 tensorflow/contrib/memory_stats/python/ops
400 …DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/contrib/memory_stats/ops/gen_memory_s…
21 from tensorflow.contrib.memory_stats.python.ops import memory_stats_ops
62 from tensorflow.contrib import memory_stats
70 "//tensorflow/contrib/memory_stats:memory_stats_py",
69 MemoryStats memory_stats = 12; field
368 cm->RecordMemoryStats(node, stats.memory_stats()); in BuildCostModel()