Add Node::IsArg() and Node::IsRetval() methods and use them

As functions become more prominent, we have been directly checking
for op types in many places. This is slow and buggy because not all
places were updated for Device* versions of _Arg and _Retval.

PiperOrigin-RevId: 238044205