我们在A100机器上运行TensorFlow,我们的模型目前非常快地消耗TFRecord。我们发现反序列化varint64是一个热点，所以我想将其从varint64替换为fixed64。我发现：
- TensorFlow的feature.proto中不支持fixed64。
- 我有一个测试，fixed64比varint64快(10倍)。
有没有方法可以在不修改TensorFlow代码的情况下实现fixed64?
我们在A100机器上运行TensorFlow,我们的模型目前非常快地消耗TFRecord。我们发现反序列化varint64是一个热点，所以我想将其从varint64替换为fixed64。我发现：
4条答案
按热度按时间nimxete21#
我们的一些机密数据需要转换为hash64,而解析varint64对hash64不友好。
nhjlsmyf2#
现在TF不支持Fixed64,我们有没有计划在反序列化hash64时添加fixed64类型？
0pizxfdo3#
如果你认为这部分是有意义的，并且你没有计划去实现这部分，我可以贡献这部分代码。
ua4mk5z44#
@zhaozheng09 贡献总是受欢迎的。请随意提交PR。谢谢！