davidstutz / tensorflow-cpp-op-exampleLinks
Simple example of implementing a new Tensorflow operation and its gradient in C++.
☆56Updated 6 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
Sorting:
- A CUDA implementation of the ZeroOut tensorflow custom op, just for fun☆11Updated 8 years ago
- Examples of C extensions for PyTorch☆256Updated 2 years ago
- Use TensorFlow efficiently☆95Updated 4 years ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 7 years ago
- ☆23Updated 9 years ago
- ☆45Updated 2 years ago
- ☆57Updated 7 years ago
- My solutions to Udacity's Parallel Programming course (CS 344)☆75Updated 8 years ago
- This project is a simple deep neural network trained using only TensorFlow C++.☆117Updated last year
- DSD model zoo. Better accuracy models from DSD training on Imagenet with same model architecture☆113Updated 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
- 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
- Training deep neural networks with low precision multiplications☆64Updated 10 years ago
- Direct C++ Interface to PyTorch☆81Updated 7 years ago
- C++ interface for mxnet☆115Updated 8 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 5 years ago
- Caffe for Sparse Convolutional Neural Network☆238Updated 2 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆268Updated 2 years ago
- PyTorch 1.0 inference in C++ on Windows10 platforms☆89Updated 6 years ago
- Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM☆49Updated 7 years ago
- Caffe2 Tutorials☆123Updated 6 years ago
- residual-SqueezeNet☆155Updated 6 years ago
- ☆47Updated 5 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆307Updated 8 years ago
- ☆22Updated 6 years ago
- ☆138Updated 8 years ago
- ☆62Updated 7 years ago
- ☆54Updated 7 years ago
- Net-Trim: Convex Pruning of Deep Neural Networks with Performance Guarantee☆60Updated 7 years ago
- ☆38Updated 8 years ago