Searched refs:GraphMemory (Results 1 – 5 of 5) sorted by relevance
51 GraphMemory memory(item); in TEST_F()54 const GraphMemory::MemoryUsage& mem_usage = in TEST_F()79 GraphMemory memory(item); in TEST_F()84 const GraphMemory::MemoryUsage& mem_usage = in TEST_F()106 GraphMemory memory(item); in TEST_F()110 const GraphMemory::MemoryUsage& cpu_mem = memory.GetPeakMemoryUsage("/CPU:0"); in TEST_F()123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F()145 GraphMemory memory(item); in TEST_F()149 const GraphMemory::MemoryUsage& gpu_mem = in TEST_F()173 GraphMemory memory(item); in TEST_F()[all …]
34 Status GraphMemory::InferStatically( in InferStatically()51 Status GraphMemory::InferDynamically(Cluster* cluster) { in InferDynamically()64 int64 GraphMemory::GetWorstCaseMemoryUsage() const { in GetWorstCaseMemoryUsage()72 void GraphMemory::InferMemUsageForNodes( in InferMemUsageForNodes()99 int64 GraphMemory::InferMemUsageForNeighbors( in InferMemUsageForNeighbors()122 static GraphMemory::LiveTensor* FindOrCreateLiveTensor( in FindOrCreateLiveTensor()124 std::unordered_map<string, GraphMemory::LiveTensor*>* live_tensors, in FindOrCreateLiveTensor()125 std::deque<GraphMemory::LiveTensor>* device_tensors) { in FindOrCreateLiveTensor()127 GraphMemory::LiveTensor* live; in FindOrCreateLiveTensor()130 GraphMemory::LiveTensor temp; in FindOrCreateLiveTensor()[all …]
29 class GraphMemory {43 explicit GraphMemory(const GrapplerItem& item) in GraphMemory() function
394 tensorflow::grappler::GraphMemory memory(*item); in TF_DeterminePeakMemoryUsage()410 const tensorflow::grappler::GraphMemory::MemoryUsage& usage = in TF_DeterminePeakMemoryUsage()
535 GraphMemory memory(*item); in SchedulingPass()552 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in SchedulingPass()952 GraphMemory memory(*item); in IdentifySwappingCandidates()972 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in IdentifySwappingCandidates()