Searched defs:wrapper_fn (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_higher_order_ops.py | 2367 def wrapper_fn(x): function
2947 def wrapper_fn(x): function
3077 def wrapper_fn(x, y): function
3215 def wrapper_fn(x): function
3238 def wrapper_fn(x): function
3315 def wrapper_fn(x, y): function
3395 def wrapper_fn(x, y): function
3477 def wrapper_fn(x): function
3503 def wrapper_fn(x, v): function
3548 def wrapper_fn(x, v): function
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | eager_transforms.py | 602 def wrapper_fn(*args): function
1290 def wrapper_fn(*args): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/ |
| H A D | mixed_precision.py | 36 def register_loss_scale_wrapper(optimizer_cls, wrapper_fn, wrapper_cls=None): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 1144 def wrapper_fn(index, reduction_index): function
1180 def wrapper_fn(merged_index, new_reduction_index): function
1758 def wrapper_fn(idx, reduction_idx): function
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 5184 def wrapper_fn(x, y): function
5192 def wrapper_fn(x, y): function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx.py | 142 def wrapper_fn(x): function
|