pytorch/torch/_prims
2022-12-08 00:17:39 +00:00
..
__init__.py [primTorch] Add prim and ref for as_strided_scatter (#88426) 2022-12-08 00:17:39 +00:00
context.py Add factory functions to python frontend (#89230) 2022-12-06 07:16:21 +00:00
executor.py Add config to PrimTorch's nvFuser executor (#84482) 2022-09-09 07:58:21 +00:00
nvfuser_executor.py Use fx.replace_pattern for removing empty_like+fill in nvFuser+PrimTorch execution (#89132) 2022-11-24 09:37:10 +00:00
nvfuser_prims.py Add factory functions to python frontend (#89230) 2022-12-06 07:16:21 +00:00