I can confirm that pytorch_android_lite:1.9.0 does not have this op (without selective build). But it works in the new 1.10 release. Just change the dependency to
implementation 'org.pytorch:pytorch_android_lite:1.10.0'
implementation 'org.pytorch:pytorch_android_torchvision_lite:1.10.0'