pytorch/caffe2
kshitij12345 c7787f7fbf [numpy compatibility]Fix argmin/argmax when multiple max/min values (#42004)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/41998
Fixes https://github.com/pytorch/pytorch/issues/22853

Pull Request resolved: https://github.com/pytorch/pytorch/pull/42004

Reviewed By: ngimel

Differential Revision: D23049003

Pulled By: mruberry

fbshipit-source-id: a6fddbadfec4b8696730550859395ce4f0cf50d6
2020-08-28 06:42:42 -07:00
..
contrib match batchmatmul on 1.0.0.6 (#43559) 2020-08-26 14:35:31 -07:00
core [pytorch] deprecate static dispatch (#43564) 2020-08-27 14:52:48 -07:00
cuda_rtc
db
distributed [caffe2] add type annotations for caffe2.distributed.python 2020-08-07 13:12:53 -07:00
experiments remediation of S205607 2020-07-17 17:19:47 -07:00
ideep Ensure IDEEP transpose operator works correctly 2020-08-11 12:58:31 -07:00
image
mobile
mpi [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -07:00
observers [codemod][caffe2] Run clang-format - 5/7 2020-06-30 15:45:11 -07:00
onnx Add python inferface to create OfflineTensor (#42516) 2020-08-04 01:31:34 -07:00
operators Reduce amount of work done within a global lock within ParallelLoadOp (#43508) 2020-08-26 18:19:40 -07:00
opt Workaround bugs in user side embedding meta info and better msgs (#43355) 2020-08-21 17:18:51 -07:00
perfkernels remediation of S205607 2020-07-17 17:19:47 -07:00
predictor [caffe2] Special handling of If/AsyncIf op in RemoveOpsByType (#42286) 2020-07-30 02:02:20 -07:00
proto [blob reorder] Seperate user embeddings and ad embeddings in large model loading script 2020-08-06 23:54:03 -07:00
python [numpy compatibility]Fix argmin/argmax when multiple max/min values (#42004) 2020-08-28 06:42:42 -07:00
quantization Fix PackedGemmMatrixFP16 repacking (#43320) 2020-08-21 10:58:18 -07:00
queue
serialize [serialize] Expose zip file alignment calculation functions (#43531) 2020-08-25 02:32:58 -07:00
sgd [caffe2] Fix the timeout (stuck) issues of dedup SparseAdagrad C2 kernel 2020-08-07 18:42:36 -07:00
share Update pthreadpool to pthreadpool:029c88620802e1361ccf41d1970bd5b07fd6b7bb. (#40524) 2020-07-09 10:00:36 -07:00
test
transforms Replaced whitelisted with allowed (#41867) 2020-07-23 16:53:51 -07:00
utils [caffe2][redo] Reimplement RemoveOpsByType with SSA (#41606) 2020-07-17 16:06:43 -07:00
video [ROCm][Caffe2] Enable MIOpen 3D Pooling (#38260) 2020-07-08 17:42:55 -07:00
.clang-format
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
c2_aten_srcs.bzl
CMakeLists.txt Add lite SequentialSampler to torch mobile (#43299) 2020-08-24 09:45:24 -07:00
README.md Grammar patch 1 (.md) (#41599) 2020-07-20 10:19:08 -07:00
release-notes.md
requirements.txt
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