Hi @parthsharma, I get those warnings too (at the start of training) but I also get a message: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:112] Plugin optimizer for device_type GPU is enabled. Training then continues using the GPU. I expect that the warning messages could be suppressed but I have not bothered to do so.