pytorch/test/lazy
Bin Bao 29189d2ba8 [LT] Add IR resuing support for manually-implemented ops
Summary: Add CanBeReused methods for manually-implemented ops and replace MakeNode with
ReuseOrMakeNode.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77616

Approved by: https://github.com/JackCaoG, https://github.com/wconstab
2022-05-26 16:04:47 +00:00
..
__init__.py
test_bindings.py
test_extract_compiled_graph.py Dynamo+LTC: handle inplace ops (#75359) 2022-04-08 03:14:28 +00:00
test_reuse_ir.py [LT] Add IR resuing support for manually-implemented ops 2022-05-26 16:04:47 +00:00
test_ts_opinfo.py Adding support for nonzero in LazyTensor shape functions 2022-05-18 17:32:48 +00:00