Searched refs:memory_stats (Results 1 – 6 of 6) sorted by relevance
138 step_stat.memory_stats().temp_memory_size()); in AddMemoryStats()141 step_stat.memory_stats().persistent_memory_size()); in AddMemoryStats()145 step_stat.memory_stats().temp_memory_size()); in AddMemoryStats()148 step_stat.memory_stats().persistent_memory_size()); in AddMemoryStats()
808 std::unique_ptr<PartitionBucketMemoryStats[]> memory_stats; in DumpStats() local810 memory_stats = std::unique_ptr<PartitionBucketMemoryStats[]>( in DumpStats()828 memory_stats[i] = bucket_stats; in DumpStats()830 memory_stats[i].is_valid = false; in DumpStats()838 if (memory_stats[i].is_valid) in DumpStats()839 dumper->PartitionsDumpBucketStats(partition_name, &memory_stats[i]); in DumpStats()
323 const MemoryStats& memory_stats) { in RecordMemoryStats() argument326 max_mem_usage_[id].temp_memory_size = memory_stats.temp_memory_size(); in RecordMemoryStats()328 memory_stats.persistent_memory_size(); in RecordMemoryStats()329 for (int64 alloc_id : memory_stats.persistent_tensor_alloc_ids()) { in RecordMemoryStats()
148 void RecordMemoryStats(const Node* node, const MemoryStats& memory_stats);
71 MemoryStats memory_stats = 12; field
366 cm->RecordMemoryStats(node, stats.memory_stats()); in BuildCostModel()