pytorch/caffe2
Chunli Fu 7929bc76a0 [shape inference] Fix dim type for Cast
Summary: ATT

Test Plan: unit test

Reviewed By: yinghai

Differential Revision: D27904584

fbshipit-source-id: b62d2eb5da0be79091c82e6300dd0c075a0bf2fe
2021-04-21 03:21:56 -07:00
..
contrib s/AutoNonVariableTypeMode/AutoDispatchBelowAutograd/ (#56423) 2021-04-20 17:17:46 -07:00
core [StaticRuntime] Replace StorageImpl with TensorImpl in MemoryPlanner (#56447) 2021-04-20 23:04:01 -07:00
cuda_rtc
db
distributed Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
experiments Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
ideep Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
image Check kernel launches in caffe2/caffe2/image (#52173) 2021-02-12 10:11:41 -08:00
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 Lint trailing newlines (#54737) 2021-03-30 13:09:52 -07:00
operators Remove an unused variable in SoftmaxWithLossOp (#56321) 2021-04-20 09:18:08 -07:00
opt [shape inference] Fix dim type for Cast 2021-04-21 03:21:56 -07:00
perfkernels [caffe2] Fix -Wundef 2021-03-31 22:24:30 -07:00
predictor Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
proto [typing] make mypy-protobuf output compatible with pyre for caffe2 type stubs (#55294) 2021-04-05 03:23:31 -07:00
python [caffe2] Speed up remote net loading 2021-04-20 22:32:40 -07:00
quantization fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
queue fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
serialize [package] Correct usage of miniz API in PyTorchStreamReader (#55725) 2021-04-13 11:50:08 -07:00
sgd optimizer exploration - v1 and v2 + fix position_weighted optimizer + decoupled weight decay (#54042) 2021-03-27 23:03:29 -07:00
share [PyTorch] Remove CAFFE2_FB_LIMITED_MOBILE_CAPABILITY (#50385) 2021-01-20 10:26:54 -08:00
test
transforms fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
utils Revert caffe2 print stack traces flag (#56496) 2021-04-20 11:43:33 -07:00
video Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
.clang-format
c2_aten_srcs.bzl Move grad_mode.h/cpp to c10. (#56204) 2021-04-16 11:50:08 -07:00
CMakeLists.txt Ignore warnings for record_function_ops (#56543) 2021-04-20 16:17:30 -07:00
README.md Grammar patch 1 (.md) (#41599) 2020-07-20 10:19:08 -07:00
release-notes.md Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
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