Home
last modified time | relevance | path

Searched refs:persistent_memory_size (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h216 MemUsage() : temp_memory_size(0), persistent_memory_size(0) {} in MemUsage()
220 Bytes persistent_memory_size; member
Dcostmodel.cc319 return max_mem_usage_[id].persistent_memory_size; in PersistentMemorySize()
327 max_mem_usage_[id].persistent_memory_size = in RecordMemoryStats()
328 memory_stats.persistent_memory_size(); in RecordMemoryStats()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.cc141 step_stat.memory_stats().persistent_memory_size()); in AddMemoryStats()
148 step_stat.memory_stats().persistent_memory_size()); in AddMemoryStats()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto46 int64 persistent_memory_size = 3; field
Dcost_graph.proto51 int64 persistent_memory_size = 12; field
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc493 EXPECT_EQ(0, node.persistent_memory_size()); in TEST_F()
497 EXPECT_LE(4 * sizeof(int64), node.persistent_memory_size()); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc1267 int64 persistent_memory_size = 0; in GenerateRunMetadata() local
1269 persistent_memory_size = total_output_size; in GenerateRunMetadata()
1271 mem_stats->set_persistent_memory_size(persistent_memory_size); in GenerateRunMetadata()
Dutils.cc334 cost_node->persistent_memory_size()); in CostGraphToOpPerformanceData()