[NFC] Use compile_all_resource_ops instead of allow_resource_ops for clarity
Firstly, rename compile_resource_ops to compile_all_resource_ops to emphasize that the device registration wants us to compile all kinds of resource operations, not just resource variable ops. Secondly using op_filter.allow_resource_ops was semantically incorrect; its purpose is to disallow clustering functional while nodes with resource variable operations, while the condition that is being changed only cares about whether we're compiling for XLA_* devices. It just so happens that op_filter.allow_resource_ops and registration->compile_all_resource_ops have the same value. PiperOrigin-RevId: 238348078
Loading
Please sign in to comment