TPU-KERNEL开发参考手册
介绍
TPU架构图
TPU的内存类型
TPU的工作模式
TPU编程模型
Host端
Device端
基础定义
舍入模式
功能函数
数据同步函数
辅助函数
GDMA 操作
基础数据操作
数据类型转换与舍入操作
一元操作
二元操作
浮点二元操作
整型二元操作
比较选择函数
浮点矩阵操作
整型矩阵操作
浮点神经网络操作
定点神经网络操作
激活函数
scatter 和 gather 操作
特殊函数
量化操作
HAU 操作
TPU-KERNEL开发参考手册
»
搜索
请启用 JavaScript 以便使用搜索功能