Tensorflow Assert в оценке

Question

Как я могу использовать опцию tf.Assert в Estimator API?Обычно вы используете блок with tf.control_dependencies([assert_op]), но я не вижу, как это сделать в этом случае.