Searched refs:inner_compiler (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/dynamo/ |
| D | test_backward_higher_order_ops.py | 98 def inner_compiler(gm_, example_inputs_): function
105 gm, backend=inner_compiler, fullgraph=True, dynamic=True
155 def inner_compiler(gm_, example_inputs_): function
162 gm, backend=inner_compiler, fullgraph=True, dynamic=True
|
| /external/pytorch/test/distributed/_composable/ |
| D | test_replicate_with_compiler.py | 62 def inner_compiler(gm_, example_inputs_): function
68 gm = torch.compile(gm, fullgraph=True, backend=inner_compiler)
|
| /external/pytorch/test/inductor/ |
| D | test_compiled_autograd.py | 445 def inner_compiler(gm_, example_inputs_): function
455 return torch.compile(gm, backend=inner_compiler)
1133 def inner_compiler(gm_, example_inputs_): function
1138 gm, backend=inner_compiler, fullgraph=True, dynamic=True
|
| /external/pytorch/torch/_dynamo/ |
| D | utils.py | 2663 def inner_compiler(gm_, example_inputs_): function
2667 return torch.compile(gm, backend=inner_compiler, fullgraph=True, dynamic=True)
|