Community
cancel
Showing results for 
Search instead for 
Did you mean: 
idata
Community Manager
383 Views

Is it possible to run certain layers of a network on different devices on Tensorflow?

Hello All, Hope all are doing okay.

 

I have a problem regarding tensorflow with NCS….here is my question.

 

I have been trying to run inference on a custom Tensorflow model on Movidius Neural Compute Stick. The problem is, the final layer is unsupported by NCSDK. Since the final layer is computationally quite light, I would like to know if there is a possibility to run that layer on CPU?

 

My previous searches led me to these:

 

     

  1. Distributed Computing Tensorflow

  2.  

  3. Placing Variables on CPU??

  4.  

 

I just need to be able to run inference up to a certain layer on the NCS, then process the last layer on CPU.

 

Thanks

0 Kudos
0 Replies
Reply