How to create custom tensorflow metric for accuracy

Question

I am trying to build a custom accuracy metric as suggested in TensorFlow docs by tracking two variables count and total.

In the upda