pytorch
d3a8505e - [jit] Added a pass to transform aten::cat ops to prim::Concat op with variable number of inputs (#59881)

Commit
4 years ago
[jit] Added a pass to transform aten::cat ops to prim::Concat op with variable number of inputs (#59881) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59881 This pass is not included in the JIT flow or anywhere else at this point. The idea is, once this lands, everyone can use this to test their workflow with this transformation and once we are convinced this is useful and/or improves performance, we can include it in the appropriate workflow. Test Plan: Imported from OSS Reviewed By: mrshenli Differential Revision: D29277876 Pulled By: navahgar fbshipit-source-id: b5be7bdcc98dced59295bd7b8f6627619cb58d41
Author
Parents
Loading