diff options
| author | Adam Paszke <adam.paszke@gmail.com> | 2018-09-11 05:56:17 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-09-11 06:02:39 -0700 |
| commit | 0ddbe668cda54271aeca99adefcf052f163847eb (patch) | |
| tree | b57a2da9c6807ca7fa4436f5beffe251a7df3ebb /tools/jit/templates | |
| parent | f84693efa96317ffb3a1c68045fa924d2d6bf2b2 (diff) | |
| download | pytorch-0ddbe668cda54271aeca99adefcf052f163847eb.tar.gz pytorch-0ddbe668cda54271aeca99adefcf052f163847eb.tar.bz2 pytorch-0ddbe668cda54271aeca99adefcf052f163847eb.zip | |
Improve shape analysis to cover all most commonly used ops (#11358)
Summary:
[Here's a list](https://gist.github.com/apaszke/f0821840bdcc67a977832dc58acc1b85) of ops that are in `register_aten_ops.cpp`, but aren't supported in shape prop. Everything else should work now.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11358
Differential Revision: D9753693
Pulled By: apaszke
fbshipit-source-id: efeae0126ce16cb56b8797fc5246405588bcae3c
Diffstat (limited to 'tools/jit/templates')
| -rw-r--r-- | tools/jit/templates/register_aten_ops.cpp | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/tools/jit/templates/register_aten_ops.cpp b/tools/jit/templates/register_aten_ops.cpp index 4ac499b99f..de205de52a 100644 --- a/tools/jit/templates/register_aten_ops.cpp +++ b/tools/jit/templates/register_aten_ops.cpp @@ -50,10 +50,6 @@ std::array<bool, N> as_bool_array(at::ArrayRef<int64_t> vec) { return res; } -at::Device as_device(ArrayRef<int64_t> elements) { - return at::Device(static_cast<at::Device::Type>(elements[0]), elements[1]); -} - RegisterOperators reg({ ${constructors} }); |