Searched defs:fake_tensor_inputs (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | debugging.py | 28 def eager(gm, fake_tensor_inputs, **kwargs): argument
35 def eager_noexcept(gm, fake_tensor_inputs, **kwargs): argument
53 def pre_dispatch_eager(gm, fake_tensor_inputs, **kwargs): argument
69 def eager_debug(gm, fake_tensor_inputs, **kwargs): argument
86 def torchscript(gm, fake_tensor_inputs): argument
120 def aot_eager_decomp_partition(gm, fake_tensor_inputs, **kwargs): argument
|
| H A D | torchxla.py | 15 def openxla_eval(model, fake_tensor_inputs): argument
19 def openxla_eval_boxed(model, fake_tensor_inputs): argument
23 def xla_backend_helper(model, fake_tensor_inputs, boxed=False): argument
|