Searched refs:dynamo_model (Results 1 – 2 of 2) sorted by relevance
123 def cudagraphs(dynamo_model, dynamo_inputs): argument198 return aot_cudagraphs(dynamo_model, dynamo_inputs)
1181 dynamo_model: torch.fx.GraphModule,1200 dynamo_model,1434 dynamo_model=model_,