I am trying to use IPlugin to add a caffe custom layer in TensorRT ,I referred the *.cu code of the caffe layer to write my custom plugin,the problem is I don’t know how to represent some sentences (from *.cu )in IPlugin. For example :
Dtype* pre_gate_data = pre_gate_.mutable_gpu_data(); (in *.cu)
the pre_gate_'s definition is Blob pre_gate_; and blobs_ is vector<shared_ptr<Blob > > blobs_;
These confused me in a long time,any suggestion is great!