pytorch/caffe2
Yinghai Lu 4be4b4c8b5 Remove weight from input of onnxifi backend op (#10575)
Summary:
The ONNXIFI backend will absorb the constant weight in Conv, so we should not add it as an input. This is just a test artifacts. Note that Onnxifi transformer will do the right thing when cutting the graph to absorb the weights.

rdzhabarov
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10575

Reviewed By: houseroad

Differential Revision: D9357339

Pulled By: yinghai

fbshipit-source-id: a613fa3acafa687295312f5211f8e9d7f77b39cd
2018-08-16 10:56:25 -07:00
..
contrib build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
core Move all operator<< overloads out of the global namespace. (#10546) 2018-08-16 07:55:27 -07:00
cuda_rtc
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 Remove weight from input of onnxifi backend op (#10575) 2018-08-16 10:56:25 -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
transforms
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
.clang-format
__init__.py
CMakeLists.txt build changes to make cpu unified build working. (#10504) 2018-08-15 17:22:36 -07:00
README.md
release-notes.md
requirements.txt Add typing into caffe2 requirements.txt for USE_ATEN (#10047) 2018-07-30 20:09:21 -07:00
VERSION_NUMBER

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