Home
last modified time | relevance | path

Searched refs:dynamo_model (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Dcudagraphs.py123 def cudagraphs(dynamo_model, dynamo_inputs): argument
198 return aot_cudagraphs(dynamo_model, dynamo_inputs)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcompile_fx.py1181 dynamo_model: torch.fx.GraphModule,
1200 dynamo_model,
1434 dynamo_model=model_,