Commit Graph

2 Commits (c71728b18217de0eb56afbb0cb6321715d1b79d3)

Author SHA1 Message Date
yyp0 367d13203e [Torch] add fold logic for some ops (#3794) 2024-10-16 16:43:31 +08:00
Rob Suderman e80054a3cc
[torch] Folders for `torch.aten.*.tensor` operators [add, sub, mul] (#2878)
Simple folder for limited size aten tensor operations. This is primarily
useful for shape computation folding as they unfortunately can use
`aten` operators. Add, sub, mul are common examples of these folders.
2024-02-19 10:28:23 -08:00