Searched refs:torchviz_model (Results 1 – 1 of 1) sorted by relevance
24 def torchviz_model(args, model, inputs, rank): function108 torchviz_model(args, model, inputs, rank)