Ahmed Taei
75c2168966
Generalize PoolingOp(CUDA) to compute 1D, 2D and 3D pooling.
...
Summary: Extend MaxPooling & AvergePooling CUDA ops to compute 1D, 2D & 3D pooling.
Differential Revision: D4866699
fbshipit-source-id: 9bf2d970f2df2b87194a539fc60c07ac19fa1042
2017-04-12 09:16:45 -07:00
Ahmed Taei
09bfc8043b
Generalize PoolingOp(CPU) to compute 1D, 2D and 3D pooling.
...
Summary: Extend the op compute 1D, 2D & 3D pooling.
Differential Revision: D4828691
fbshipit-source-id: 87540e82ed20d1361476cfbc43a708de9ca7a88e
2017-04-11 18:18:21 -07:00
Yangqing Jia
91ebfa3c7c
Unit test for big batch size avg pooling
...
Summary: basically copied test_pooling and hard coded values
Reviewed By: prigoyal
Differential Revision: D4428162
fbshipit-source-id: 6c0444ac8c21f08824df7ff53999a94967607dc4
2017-01-18 19:29:20 -08:00
Yangqing Jia
5eb836880d
Add unittest.main() lines to test scripts under python/operator_test
...
Summary:
Needed by oss.
This is done by running the following line:
find . -name "*_test.py" -exec sed -i '$ a \\nif __name__ == "__main__":\n import unittest\n unittest.main()' {} \;
Reviewed By: ajtulloch
Differential Revision: D4223848
fbshipit-source-id: ef4696e9701d45962134841165c53e76a2e19233
2016-11-29 15:18:37 -08:00
Yangqing Jia
238ceab825
fbsync. TODO: check if build files need update.
2016-11-15 00:00:46 -08:00
Yangqing Jia
44509f9f91
fbsync: mostly lint changes, added mkl files
2016-10-11 22:45:06 -07:00
