tf.compat.v1.train.piecewise_constant