摘要: 常量内存是在变量前面加上 __constant__,常量内存用于保存核函数执行期间不会发生变化的数据,NVIDIA向硬件提供了 64KB 的常量内存,在通常情况下,使用常量内存代替全局内存能有效的缩减内存带宽 常量内存的赋值使用 cudaMemcpyToSymbol() 函数 示例代码中我创建了一个 阅读全文
posted @ 2023-03-28 18:11 Wangtn 阅读(597) 评论(0) 推荐(0) 编辑
摘要: checkpoint文件结构 saved_model文件结构 import tensorflow as tf def convert_model(): trained_checkpoint_prefix = '/home/tiwang/code/jupyter-notebook/DIEN/dien/ 阅读全文
posted @ 2023-03-28 14:50 Wangtn 阅读(131) 评论(0) 推荐(0) 编辑