Use cuCollections for Tensorflow model inference?

@Tim_Wolfe is there any code you can reference or share to build this?