pytorch
8564ed24 - do not need to check if element in dict input is Tensor. (#97866)

Comment changes are shownComment changes are hidden
Commit
1 year ago
do not need to check if element in dict input is Tensor. (#97866) sometimes it's a tuple with tensor element such as past value key in text generation case Fixes https://github.com/pytorch/pytorch/issues/97229 Pull Request resolved: https://github.com/pytorch/pytorch/pull/97866 Approved by: https://github.com/jgong5, https://github.com/davidberard98
Author
Committer
Parents
  • test/jit
    • File
      test_tracer.py
  • torch/csrc/jit/python
    • File
      python_tracer.cpp