-
rxsang authored
* Update resnet_model.py * Ensure static shapes when enabling XLA. * Define `drop_remainder` as a variable. * Handles per_replica_batch_size in non-XLA mode * Remove trailing whitespace.
371645fc
* Update resnet_model.py * Ensure static shapes when enabling XLA. * Define `drop_remainder` as a variable. * Handles per_replica_batch_size in non-XLA mode * Remove trailing whitespace.