pytorch/test/cpp
Mikhail Zolotukhin 85126629a5 [TensorExpr] Add support for constant tensors in tensorexpr kernel. (#56319)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56319

With this change the TorchScript graph can have constant tensors in it
and we still will be able to lower it to TE. The constants are
registered (or bound) within the `TensorExprKernel` object and when the
codegen is called, they are passed along with usual inputs and outputs.

Test Plan: Imported from OSS

Reviewed By: navahgar

Differential Revision: D27838747

Pulled By: ZolotukhinM

fbshipit-source-id: 4a519d66fcc07fe5fa53f5cf9af28d25611f8437
2021-04-17 11:15:35 -07:00
..
api Add padding_idx argument to EmbeddingBag (#49237) 2021-04-14 09:38:01 -07:00
common Trim libshm deps, move tempfile.h to c10 (#17019) 2019-02-13 19:38:35 -08:00
dist_autograd Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
jit [PyTorch] Lite interpreter with a backend delegate (#54462) 2021-04-06 00:55:26 -07:00
lite_interpreter_runtime [PyTorch] Lite interpreter with a backend delegate (#54462) 2021-04-06 00:55:26 -07:00
rpc [TensorPipe] Use Descriptor::Tensor::sourceDevice in tensorpipe_agent. (#55821) 2021-04-15 03:21:26 -07:00
tensorexpr [TensorExpr] Add support for constant tensors in tensorexpr kernel. (#56319) 2021-04-17 11:15:35 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00