pytorch/test/cpp
Bert Maher e542e67253 [nnc] Test case for computeAt with reduction (#54755)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54755

As title.  A step on the way to using computeAt to optimize
convolution.
ghstack-source-id: 125054730

Test Plan: new test

Reviewed By: ZolotukhinM

Differential Revision: D27353663

fbshipit-source-id: 930e09d96d1f74169bf148cd30fc195c6759a3e9
2021-03-26 20:25:18 -07:00
..
api Add padding='same' mode to conv{1,2,3}d (#45667) 2021-03-18 16:22:03 -07:00
common
dist_autograd Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
jit Merge CUDA Streams and Events (#53902) 2021-03-26 14:19:39 -07:00
lite_interpreter_runtime [PyTorch] Enable test_lite_interpreter_runtime running in android (#54579) 2021-03-24 14:45:27 -07:00
rpc Use type-erased union for Buffer. (#54251) 2021-03-19 04:58:09 -07:00
tensorexpr [nnc] Test case for computeAt with reduction (#54755) 2021-03-26 20:25:18 -07:00
__init__.py