davidstutz / tensorflow-cpp-op-example
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
- ☆22Updated 6 years ago
- A CUDA implementation of the ZeroOut tensorflow custom op, just for fun☆11Updated 8 years ago
- ☆45Updated 2 years ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆64Updated 7 years ago
- ☆23Updated 8 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
- Python Binding to NVRTC☆79Updated 6 months ago
- PyTorch 1.0 inference in C++ on Windows10 platforms☆89Updated 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
- ☆57Updated 6 years ago
- Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM☆49Updated 7 years ago
- This project is a simple deep neural network trained using only TensorFlow C++.☆118Updated last year
- Use TensorFlow efficiently☆95Updated 3 years ago
- ☆53Updated 7 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 6 years ago
- Examples of C extensions for PyTorch☆255Updated 2 years ago
- Efficient forward propagation for BCNNs☆50Updated 7 years ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- ☆38Updated 8 years ago
- ☆18Updated 7 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- Caffe with NNPACK integration☆58Updated 9 years ago
- Greentea LibDNN - a universal convolution implementation supporting CUDA and OpenCL☆135Updated 8 years ago
- Direct C++ Interface to PyTorch☆80Updated 6 years ago
- ☆45Updated 8 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆264Updated last year
- Some tools and examples for pyCaffe including LMDB I/O, custom Python layers and monitoring training error and loss.☆77Updated 8 years ago
- The "CUDA templates" are a collection of C++ template classes and functions which provide a consistent interface to NVIDIA's "Compute Uni…☆27Updated 13 years ago
- ☆28Updated 7 years ago