introduce flags to set omp and mkl threads (#21472)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21472
as title
Reviewed By: hl475
Differential Revision: D15695846
fbshipit-source-id: 44437f6b94a9c583275fcc711bb6ccf2b04f90fc