caffe中cpu_data 和mutable_cpu_data

来源:互联网 发布:视频服务器软件 编辑:程序博客网 时间:2024/06/16 00:11
一开始看代码的时候会纳闷,为啥caffe里头又一个
cpu_data还要有一个mutable_cpu_data

[cpp] view plain copy
 在CODE上查看代码片派生到我的代码片
  1. const void* SyncedMemory::cpu_data() {  
  2.   to_cpu();  
  3.   return (const void*)cpu_ptr_;  
  4. }  

[cpp] view plain copy
 在CODE上查看代码片派生到我的代码片
  1. void* SyncedMemory::mutable_cpu_data() {  
  2.   to_cpu();  
  3.   head_ = HEAD_AT_CPU;  
  4.   return cpu_ptr_;  
  5. }  

他们之间就相差一个
head_ = HEAD_AT_CPU;

当你想读取数据的时候请使用cpu_data
当然想修改数据的时候请你使用mutable_cpu_data。
这样就提示系统数据我改过啦,你要小心了的意思(我只知道数据一定在CPU上)。

0 0
原创粉丝点击