Searched refs:mutable_memory_stats (Results 1 – 3 of 3) sorted by relevance
153 auto* ms = stats_->mutable_memory_stats(); in SetMemory()
1024 ::tensorflow::MemoryStats* mutable_memory_stats(); in Swap()2406 inline ::tensorflow::MemoryStats* NodeExecStats::mutable_memory_stats() { in mutable_memory_stats() function
1318 auto* mem_stats = node_stats->mutable_memory_stats(); in GenerateRunMetadata()