Searched defs:impl_aten (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | inductor_prims.py | 16 impl_aten, argument
|
| /external/pytorch/torch/_prims/ |
| D | rng_prims.py | 28 def register_rng_prim(name, schema, impl_aten, impl_meta, doc, tags=None): argument
|
| D | __init__.py | 2395 def _make_reduction_prim(name: str, impl_aten, doc): argument
2406 def _make_var_reduction_prim(name: str, impl_aten, doc): argument
|