use shallow for cub (#4807)
use shallow for cub (#4807)


Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet,Guneet-Dhillon/mxnet