I found that the answer is to make sure you are assigning your custom UIInputView directly to the UIInputViewController.inputView property, rather than adding it as a subview of .inputView.
See my slightly longer answer here:
https://developer.apple.com/forums/thread/778827
Hope this helps someone!
Topic:
UI Frameworks
SubTopic:
UIKit
Tags: