Majority of Tensorflow models are NHWC and the GPU is good with NCHW. I am working on some changes to frontend to enable GPU by converting operators to NCHW.
This is the related discussion for the same.
Majority of Tensorflow models are NHWC and the GPU is good with NCHW. I am working on some changes to frontend to enable GPU by converting operators to NCHW.
This is the related discussion for the same.