Thanks for the reply @ptrblck, but unfortunately there is no use of 3rd party libraries. In fact, it is a new neural processing unit that performs operations such as convolution in hardware and, in this case, constants and placeholders must be in NCHW layout also.