pytorch
2130f4cc - Use c10::ArrayRef instead of std::vector for the jit::unpickle's tensor_table. (#54428)

Commit
3 years ago
Use c10::ArrayRef instead of std::vector for the jit::unpickle's tensor_table. (#54428) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54428 Using c10::ArrayRef as the parameter type makes the API more flexible and allows the caller to leverage small-buffer optimizations (e.g. c10::SmallVector, std::array) for performance critical cases. Test Plan: No behavioral changes. Run the existing unit and integration tests. Reviewed By: suo Differential Revision: D27232222 fbshipit-source-id: 7b13bc6bd02257097ca119077028fbccc68cc925
Author
Parents
Loading