Showing results for 
Show  only  | Search instead for 
Did you mean: 
Registered: ‎05-14-2020

How to move a Caffe Layer to be processed in the CPU?

I am using vai_c_caffe but it doesn't support my layer (Reshape Layer). In the forum they say that I have to move the layer to be processed in the CPU, but I didn't find an example how to do that. Do you know where I can find an example? or explains me the process how to approach my problem?

Thanks in advance

layer {
    name: "reshape"
    type: "Reshape"
    bottom: "data"
    top: "data_reshape"
    reshape_param {
      shape {
        dim: 0  # copy the dimension from below
        dim: 1
        dim: -1
        dim: 3 # infer it from the other dimensions

That was my orignally post to this topic:

0 Kudos
0 Replies