tf.debugging.assert_same_float_dtype