Python
Java
PHP
IOS
Android
Nodejs
JavaScript
Html5
Windows
Ubuntu
Linux
OneFlow 中的 Softmax
Softmax 是深度学习模型中的常见算子 PyTorch 的 Softmax 算子直接调用 cuDNN 的接口 而 OneFlow 内部针对输入数据的类别数量 采用3个 kernel 来分别处理 在多数情况下都可以获得比 cuDNN 更优
oneflow
GPU
deeplearning
人工智能
【CUDA学习笔记】OneFlow公众号CUDA算子优化文章学习笔记
1 CUDA学习资料合集 OneFlow 岁末年初 为你打包了一份技术合订本 2 GPU概念介绍 GPU的硬件结构与执行原理 开源100天 OneFlow送上 百天大礼包 深度学习框架如何进行性能优化 2 1 内存模型 2 1 1 Bank
学习
oneflow
人工智能
编译 OneFlow 模型
本篇文章译自英文文档 Compile OneFlow Models tvm 0 14 dev0 documentation 作者是 BBuf Xiaoyu Zhang GitHub 更多 TVM 中文文档可访问 Apache TVM 是一个
TVM
oneflow
深度学习
AI 编译器