Hello!
Yes, between the R-series Toyota transmissions, the input shafts are definitely interchangeable. For example, the shorter R154 and 3.0-liter R150F input shafts are sometimes needed to replace the longer input shaft in the 3.4-liter R150F.
Thank you,
Kyle