davidstutz / tensorflow-cpp-op-example
Simple example of implementing a new Tensorflow operation and its gradient in C++.
☆56Updated 5 years ago
Alternatives and similar repositories for tensorflow-cpp-op-example:
Users that are interested in tensorflow-cpp-op-example are comparing it to the libraries listed below
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 7 years ago
- ☆45Updated 2 years ago
- ☆22Updated 6 years ago
- A CUDA implementation of the ZeroOut tensorflow custom op, just for fun☆11Updated 7 years ago
- This is my original repository of the decaf code. Decaf is a precursor of Caffe written in Python for deep image classification. It is de…☆43Updated 11 years ago
- ☆23Updated 8 years ago
- Direct C++ Interface to PyTorch☆80Updated 6 years ago
- Caffe with NNPACK integration☆59Updated 8 years ago
- Use TensorFlow efficiently☆95Updated 3 years ago
- ☆58Updated 6 years ago
- PyTorch 1.0 inference in C++ on Windows10 platforms☆89Updated 6 years ago
- This is a demo project that shows how you can utilize Caffe2's modular design and build a library on top of it.☆40Updated 5 years ago
- ☆53Updated 6 years ago
- Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM☆49Updated 7 years ago
- Python Binding to NVRTC☆79Updated 3 months ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- Net-Trim: Convex Pruning of Deep Neural Networks with Performance Guarantee☆59Updated 6 years ago
- ☆45Updated 7 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- SqueezeNet Generator☆32Updated 6 years ago
- ☆28Updated 7 years ago
- Examples of C extensions for PyTorch☆257Updated last year
- mobilenet-mxnet☆145Updated 6 years ago
- This project is a simple deep neural network trained using only TensorFlow C++.☆118Updated last year
- ☆63Updated 6 years ago
- This is an example of how to train a MNIST network in Python and run it in c++ with pytorch 1.0☆96Updated 6 years ago
- residual-SqueezeNet☆155Updated 5 years ago
- mpi-caffe☆49Updated 5 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆29Updated 8 years ago
- Some tools and examples for pyCaffe including LMDB I/O, custom Python layers and monitoring training error and loss.☆77Updated 7 years ago