Register DEVICE_DEFAULT for Assign(Add/Sub)VariableOp
Ran into this when trying to run this example (https://www.tensorflow.org/agents/tutorials/6_reinforce_tutorial) with the example Pluggable Device.