TensorRT assert im.device.type != 'cpu'
on export
#6340
Merged
assert im.device.type != 'cpu'
on export
#6340