Searched refs:GraphMemory (Results 1 – 6 of 6) 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()63 int64 GraphMemory::GetWorstCaseMemoryUsage() const { in GetWorstCaseMemoryUsage()71 void GraphMemory::InferMemUsageForNodes( in InferMemUsageForNodes()98 int64 GraphMemory::InferMemUsageForNeighbors( in InferMemUsageForNeighbors()121 static GraphMemory::LiveTensor* FindOrCreateLiveTensor( in FindOrCreateLiveTensor()123 std::unordered_map<string, GraphMemory::LiveTensor*>* live_tensors, in FindOrCreateLiveTensor()124 std::deque<GraphMemory::LiveTensor>* device_tensors) { in FindOrCreateLiveTensor()126 GraphMemory::LiveTensor* live; in FindOrCreateLiveTensor()129 GraphMemory::LiveTensor temp; in FindOrCreateLiveTensor()[all …]
29 class GraphMemory {43 explicit GraphMemory(const GrapplerItem& item) in GraphMemory() function
515 bool SchedulingPass(Cluster* cluster, std::unique_ptr<GraphMemory>* memory_ptr, in SchedulingPass()543 memory_ptr->reset(new GraphMemory(*item)); in SchedulingPass()551 const GraphMemory& memory = **memory_ptr; in SchedulingPass()561 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in SchedulingPass()982 std::unique_ptr<GraphMemory>* memory_ptr, in IdentifySwappingCandidates()986 memory_ptr->reset(new GraphMemory(*item)); in IdentifySwappingCandidates()994 const GraphMemory& memory = **memory_ptr; in IdentifySwappingCandidates()1007 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in IdentifySwappingCandidates()1150 Cluster* cluster, std::unique_ptr<GraphMemory>* memory, in SwappingPass()1408 std::unique_ptr<GraphMemory> memory; in Optimize()
306 tensorflow::grappler::GraphMemory memory(*item); in PYBIND11_MODULE()319 const tensorflow::grappler::GraphMemory::MemoryUsage& usage = in PYBIND11_MODULE()
308 tensorflow::grappler::GraphMemory::InferStatically309 tensorflow::grappler::GraphMemory::InferDynamically