pytorch/caffe2
Andrey Malevich bce4c82f0d [C2] Add TypeAndShape Inference logic for ReduceMean (#51828)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51828

As desc.

Test Plan: Unit-tests.

Differential Revision: D26293844

fbshipit-source-id: 2eb2a694c439b794ad7c134409e2b8926aabc91f
2021-02-08 00:57:47 -08:00
..
contrib replace silufp16 with cubic interpolation (#51645) 2021-02-03 12:58:38 -08:00
core raise windows tol to 30% (#51733) 2021-02-04 14:09:10 -08:00
cuda_rtc
db
distributed Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
experiments
ideep Upgrade oneDNN (mkl-dnn) to v1.7 (#47853) 2020-12-03 11:54:31 -08:00
image
mobile [PyTorch Mobile] Add continuous build config for xplat/caffe2 2020-12-09 16:58:20 -08:00
mpi Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
observers Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
onnx Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
operators [C2] Add TypeAndShape Inference logic for ReduceMean (#51828) 2021-02-08 00:57:47 -08:00
opt add shape inference for Int8GenQuantParamsMinMax 2021-02-06 17:50:43 -08:00
perfkernels
predictor Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
proto Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
python Enable GPU/RE tags for caffe2/caffe2/python/TARGETS 2021-02-05 13:52:48 -08:00
quantization [p2c2] Add support for Int8FCPackWeight in model transformation 2021-01-26 16:35:23 -08:00
queue Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
serialize T66557700 Support default argument values of a method (#48863) 2021-02-01 18:35:13 -08:00
sgd [pytorch] [PR] Rename cuda kernel checks to C10 (#48615) 2020-12-01 18:19:07 -08:00
share [PyTorch] Remove CAFFE2_FB_LIMITED_MOBILE_CAPABILITY (#50385) 2021-01-20 10:26:54 -08:00
test
transforms Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
utils add and adjust kernel launch checks under fbcode/caffe2/caffe2/utils (#50862) 2021-01-21 15:20:55 -08:00
video Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
.clang-format
__init__.py
c2_aten_srcs.bzl
CMakeLists.txt Generate header with version #defines for LibTorch (#50073) 2021-02-03 22:18:53 -08:00
README.md
release-notes.md
requirements.txt Remove references to typing module from setup.py (#47677) 2020-11-12 10:04:38 -08: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/pytorch/pytorch/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai