Searched refs:create_tensor_allocation_info (Results 1 – 1 of 1) sorted by relevance
49 def create_tensor_allocation_info(graph: torch.fx.Graph) -> List[MemoryTimeline]: function130 memory_timeline = create_tensor_allocation_info(exported_program.graph)