I want to use caffe Reverse layer in Tensorrt, but I don’t find the api.
layer {
name: “reverse1”
type: “Reverse”
bottom: “lstm_input”
top: “rlstm_input”
reverse_param {
axis: 0
}
}
How can I use this layer in tensorrt, thank!
I want to use caffe Reverse layer in Tensorrt, but I don’t find the api.
layer {
name: “reverse1”
type: “Reverse”
bottom: “lstm_input”
top: “rlstm_input”
reverse_param {
axis: 0
}
}
How can I use this layer in tensorrt, thank!
Where does your Reverse layer come from? It doesn’t appear to be a standard Caffe layer. Perhaps it’s a custom layer of your own? If this is the case, you should look into using TensorRT’s Plugin model.