tensorflow / custom-op
Guide for building custom op for TensorFlow
☆378Updated last year
Related projects ⓘ
Alternatives and complementary repositories for custom-op
- A performant and modular runtime for TensorFlow☆756Updated last month
- TVM integration into PyTorch☆452Updated 4 years ago
- ☆568Updated 6 years ago
- ☆399Updated this week
- common in-memory tensor structure☆912Updated last month
- Explore the Capabilities of the TensorRT Platform☆260Updated 3 years ago
- TensorFlow-nGraph bridge☆137Updated 3 years ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆692Updated last year
- Dive into Deep Learning Compiler☆643Updated 2 years ago
- Python bindings for NVTX☆66Updated last year
- TensorFlow/TensorRT integration☆736Updated 11 months ago
- A tensor-aware point-to-point communication primitive for machine learning☆249Updated last year
- tensorflow源码阅读笔记☆189Updated 6 years ago
- ☆374Updated 7 years ago
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆946Updated last month
- Collective communications library with various primitives for multi-machine training.☆1,227Updated this week
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆42Updated last year
- To make it easy to benchmark AI accelerators☆179Updated last year
- [Deprecated] The TensorFlow Profiler (TFProf) UI provides a visual interface for profiling TensorFlow models.☆136Updated 5 years ago
- TensorFlow examples in C, C++, Go and Python without bazel but with cmake and FindTensorFlow.cmake☆439Updated 5 years ago
- A profiling and performance analysis tool for TensorFlow☆360Updated this week
- Place for meetup slides☆140Updated 4 years ago
- Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO☆706Updated this week
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆291Updated 5 years ago
- High performance Cross-platform Inference-engine, you could run Anakin on x86-cpu,arm, nv-gpu, amd-gpu,bitmain and cambricon devices.☆532Updated 2 years ago
- A GPU performance profiling tool for PyTorch models☆495Updated 3 years ago
- ☆303Updated last week
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- Symbolic Expression and Statement Module for new DSLs☆206Updated 4 years ago
- DeepLearning Framework Performance Profiling Toolkit☆277Updated 2 years ago