tf.keras.backend.is_float_dtype