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:
- Examples of C extensions for PyTorch☆256Updated 2 years ago
- A CUDA implementation of the ZeroOut tensorflow custom op, just for fun☆11Updated 8 years ago
- Use TensorFlow efficiently☆96Updated 4 years ago
- This project is a simple deep neural network trained using only TensorFlow C++.☆117Updated last year
- ☆47Updated 3 years ago
- ☆57Updated 7 years ago
- ☆22Updated 9 years ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 7 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
- ☆54Updated 7 years ago
- ☆22Updated 6 years ago
- This repository contains several tools useful for pytorch users.☆47Updated 6 years ago
- PyTorch 1.0 inference in C++ on Windows10 platforms☆89Updated 6 years ago
- DSD model zoo. Better accuracy models from DSD training on Imagenet with same model architecture☆113Updated 6 years ago
- Direct C++ Interface to PyTorch☆81Updated 7 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆268Updated 2 years ago
- C++ interface for mxnet☆115Updated 8 years ago
- Greentea LibDNN - a universal convolution implementation supporting CUDA and OpenCL☆137Updated 8 years ago
- Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM☆49Updated 7 years ago
- Caffe for Sparse Convolutional Neural Network☆237Updated 2 years ago
- ☆62Updated 7 years ago
- Image classification C++ example using tensorflow shared library☆104Updated 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 12 years ago
- ☆18Updated 7 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆41Updated 6 years ago
- Caffe with NNPACK integration☆59Updated 9 years ago
- Training deep neural networks with low precision multiplications☆64Updated 10 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆307Updated 8 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 5 years ago
- Efficient forward propagation for BCNNs☆49Updated 8 years ago