mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Kittipat Virochsiri fd2aa93b37 Exposing LengthsSum/Mean/Max in pytorch (#20802 )

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20802

Need this for sequence model

Reviewed By: dzhulgakov

Differential Revision: D15448529

fbshipit-source-id: cd5abe3b689fc0e02feff10faf8cd61c99369f4f

2019-05-22 13:55:19 -07:00

contrib

Enable ROCm multi-gpu with Gloo

2019-05-07 09:55:47 -07:00

core

Add onnxifi support for Int8FCDNNLowPPackedWeightBlob (#20564 )

2019-05-20 16:57:11 -07:00

cuda_rtc

Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764 )

2019-03-07 18:38:53 -08:00

Apply modernize-use-override (2nd iteration)

2019-02-14 16:52:57 -08:00

distributed

Manual hipify caffe2/distributed and rocm update (no hcc modules support) (#18088 )

2019-03-29 11:07:32 -07:00

experiments

Tensor construction codemod(ResizeLike) - 1/7 (#15073 )

2018-12-19 21:38:48 -08:00

ideep

Remove useless input shape checker in conv (#19608 )

2019-04-24 11:39:43 -07:00

image

Clang-format ImageInputOp (#20441 )

2019-05-16 23:00:09 -07:00

mobile

update legacy plus one for mpscnn

2019-05-16 18:17:18 -07:00

mpi

Tensor construction codemod(ResizeLike) - 3/7 (#15122 )

2018-12-14 02:08:37 -08:00

observers

Remove GPU dependency from ProfileObserver (#17592 )

2019-03-04 10:00:46 -08:00

onnx

Export randn_like in ONNX exporter (#20093 )

2019-05-06 14:54:46 -07:00

operators

Exposing LengthsSum/Mean/Max in pytorch (#20802 )

2019-05-22 13:55:19 -07:00

opt

Add onnxifi support for Int8FCDNNLowPPackedWeightBlob (#20564 )

2019-05-20 16:57:11 -07:00

perfkernels

eliminate FE_INVALID exceptions related to fp16 conversion (#20390 )

2019-05-13 23:42:01 -07:00

predictor

Removing cyclic dependency (#20511 )

2019-05-14 18:55:19 -07:00

proto

Add support for __getstate__/__setstate__ on module (#20242 )

2019-05-17 14:43:14 -07:00

python

Exposing LengthsSum/Mean/Max in pytorch (#20802 )

2019-05-22 13:55:19 -07:00

quantization

Add onnxifi support for Int8FCDNNLowPPackedWeightBlob (#20564 )

2019-05-20 16:57:11 -07:00

queue

Revert "Tensor construction codemod(raw_mutable_data) (#16373 )" (#18680 )

2019-04-01 14:39:13 -07:00

serialize

Add the support of feature store example in pytorch model in fblearner (#20040 )

2019-05-20 12:58:27 -07:00

sgd

shape inference for learning rate op (#20020 )

2019-05-14 23:34:32 -07:00

Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764 )

2019-03-07 18:38:53 -08:00

test

move flags to c10 (#12144 )

2018-10-04 02:09:56 -07:00

transforms

fix -Wsign-compare warnings for some files inside c2 (#18123 )

2019-03-19 10:39:20 -07:00

utils

Remove unused var (ws_) and use vars in undefined case for compile (#20667 )

2019-05-20 13:52:57 -07:00

video

codemode change missing [from D13586737]

2019-05-13 20:44:04 -07:00

__init__.py

Experimental support for setup.py develop mode install

2018-02-12 23:36:18 -08:00

.clang-format

Add clang-format style file to caffe2

2018-02-05 20:35:18 -08:00

CMakeLists.txt

Revert D15248618: Split ATen/Parallel into interface and backend

2019-05-22 09:55:51 -07:00

README.md

Fix issues link in Caffe2 readme (#10711 )

2018-08-20 16:55:11 -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

move flags to c10 (#12144 )

2018-10-04 02:09:56 -07:00

README.md

Caffe2

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.

README.md

Caffe2

Questions and Feedback

Further Resources on Caffe2.ai