cancel
Showing results for 
Show  only  | Search instead for 
Did you mean: 
noah98
Observer
Observer
197 Views
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:

https://forums.xilinx.com/t5/AI-and-Vitis-AI/vai-c-caffe-fails-to-compile-Reshape-Layer/m-p/1138589#M5190

0 Kudos
0 Replies