pytorch/caffe2
Junjie Bai 84427d26db Add aten_op to caffe2 onnx (python) backend
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/10579

Reviewed By: houseroad

Differential Revision: D9357837

fbshipit-source-id: 2cc6fedbaf088df7e11b52a91dfe3b8f0d7fd599
2018-08-16 00:39:30 -07:00
..
contrib build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
core Resolving error C2487 "member of dll interface class may not be declared with dll interface" by removing nested CAFFE2_API. (#10572) 2018-08-16 00:25:41 -07:00
cuda_rtc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
db Add db directory for hipifying (#10428) 2018-08-13 10:24:56 -07:00
distributed build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
experiments Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
ideep Add new MKLDNN fallback operators (#10526) 2018-08-15 15:55:22 -07:00
image Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
mkl s/DISABLE_COPY_AND_ASSIGN/AT_DISABLE_COPY_AND_ASSIGN/ (#10275) 2018-08-07 08:54:26 -07:00
mobile Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
mpi IsType<TensorCPU> -> IsType<Tensor>(CPU) (#10135) 2018-08-03 17:24:59 -07:00
observers IsType<TensorCPU> -> IsType<Tensor>(CPU) (#10135) 2018-08-03 17:24:59 -07:00
onnx Handle the "spatial" attribute in onnx BatchNormalization op (#9492) 2018-07-27 22:09:15 -07:00
operators Update Im2Col related to make preparation for group conv in NHWC order. (#10439) 2018-08-15 17:10:24 -07:00
opt Fold AffineChannel to Conv, the same way as BN (for Detectron models) (#10293) 2018-08-13 22:43:37 -07:00
perfkernels Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
predictor build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
proto Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
python Add aten_op to caffe2 onnx (python) backend 2018-08-16 00:39:30 -07:00
queue build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
sgd Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
share Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
test Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
transforms [codemod][caffe2] comment out unused parameters 2018-03-30 21:00:44 -07:00
utils build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
video Remove caffe1 specific proto (#10380) 2018-08-10 11:10:26 -07:00
__init__.py
.clang-format
CMakeLists.txt build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
README.md Move README.md to caffe2/ in prep for merge. (#2479) 2018-03-29 14:09:32 -07:00
release-notes.md Cleanup of the shipit commit (#8956) 2018-06-27 14:41:59 -07:00
requirements.txt Add typing into caffe2 requirements.txt for USE_ATEN (#10047) 2018-07-30 20:09:21 -07:00
VERSION_NUMBER Rename setup.py to setup_caffe2.py (#2483) 2018-03-30 07:29:55 -07:00

Caffe2

Jenkins Build Status

Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.

Questions and Feedback

Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai