saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2
History
Kittipat Virochsiri 2b134c72e6 Add interface to provide blob types to shape&type inference (#9643) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9643

Current map interface assumes float data type, which is not always correct.

Reviewed By: kennyhorror

Differential Revision: D8455784

fbshipit-source-id: b94a31267760f7f97c15aa4b03008affc347fd10
2018-07-24 11:58:05 -07:00
..
contrib Update onnx-tensort module to the latest (#9469) 2018-07-16 17:10:16 -07:00
core Add interface to provide blob types to shape&type inference (#9643) 2018-07-24 11:58:05 -07:00
cuda_rtc
db
distributed HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
experiments Add random data filler to predictor bench to support production nets (#9520) 2018-07-18 00:46:02 -07:00
ideep Add fallback to TensorCPU if there are unsupported types for IDEEP Tensor (#9667) 2018-07-23 13:54:57 -07:00
image HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
mkl Update channel_shuffle_op and transpose 2d to speed up ShuffleNet (#9525) 2018-07-21 12:54:33 -07:00
mobile Fixes xcode 10 beta 4 compile error (#9748) 2018-07-24 11:11:35 -07:00
mpi
observers
onnx Adapt OnnxifiOp to removed suffix handling in ONNXIFI loader (#9571) 2018-07-18 19:26:23 -07:00
operators Add interface to provide blob types to shape&type inference (#9643) 2018-07-24 11:58:05 -07:00
opt Allow multiple ops.def and clean up code gen in general 2018-07-23 15:44:04 -07:00
perfkernels Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
proto Adds serialization path (#9035) 2018-07-02 09:09:39 -07:00
python Add interface to provide blob types to shape&type inference (#9643) 2018-07-24 11:58:05 -07:00
queue Eanble python tests on ROCM (#9616) 2018-07-24 11:37:58 -07:00
sgd HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
share Simplify pthreadpool implementation on top of Caffe2 thread pool (#7666) 2018-06-18 19:06:50 -07:00
test
transforms
utils Eanble python tests on ROCM (#9616) 2018-07-24 11:37:58 -07:00
video
.clang-format
__init__.py
CMakeLists.txt HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
README.md
release-notes.md Cleanup of the shipit commit (#8956) 2018-06-27 14:41:59 -07:00
requirements.txt Add pyHIPIFY scripts needed for ROCm transpilation to PyTorch (#8812) 2018-07-10 18:02:43 -07:00
VERSION_NUMBER

README.md

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