Google Git
Sign in
android / platform / external / pytorch / da72658fa8
commitda72658fa84deff9414da500cec4265145ae4a50[log] [tgz]
authorMartin Raison <raison@fb.com>Tue Nov 29 06:20:40 2016 -0800
committerBram Wasti <bwasti@dev11999.prn1.facebook.com>Tue Nov 29 15:18:39 2016 -0800
tree4330f38db06b3f069a30a40d56c7c16bb4e5e446
parentf16c2fe3da00d48d2dd54c87a6cf2a32523403d1 [diff]
sparsehash-based implementation of UniqueOp

Summary:
Faster implementation of UniqueOp using google::dense_hash_map, as suggested by dzhulgakov. I haven't benchmarked it precisely but early measurements with my workflow show a significant speed bump (this operation went from using 20% of overall CPU time down to 7%).

I gated the implementation using the "engine" feature, to avoid adding sparsehash as a dependency to caffe2.

Reviewed By: dzhulgakov

Differential Revision: D4219768

fbshipit-source-id: 2f142981e772105b42fffa24afb199ef816f8e0c
1 file changed
tree: 4330f38db06b3f069a30a40d56c7c16bb4e5e446
  1. caffe/
  2. caffe2/
  3. docs/
  4. third_party/
  5. .Doxyfile
  6. .gitignore
  7. .gitmodules
  8. build.py
  9. build_android.py
  10. build_android_prepare.py
  11. LICENSE
  12. Makefile
  13. README.md
README.md

Caffe2

Caffe2 is a deep learning framework made with expression, speed, and modularity in mind. It is an experimental refactoring of Caffe, and allows a more flexible way to organize computation.

Read the installation instructions for installation details.

License and Citation

Caffe2 is released under the BSD 2-Clause license.