Loop partitioning and Tensorization - Work on different IR levels

[janimesh]

@yzhliu should be able to answer this.