Linked twist map formalism in two and three dimensions applied to mixing in tumbled granular flows