def Combined(x): ys = [s.Value(x) for s in self.schedules] return tf.reduce_min(tf.stack(ys), axis=0)
def Value(self): ys = [s.Value() for s in self.schedules] return tf.reduce_min(tf.stack(ys), axis=0)
def KeyFunc(batch): key = tf.reduce_min(batch.bucket_keys) idx = tf.reduce_sum( tf.cast(tf.greater(key, p.bucket_upper_bound), tf.int32)) return tf.constant(p.bucket_upper_bound, dtype=tf.int64)[idx]
def Value(self, step=None): ys = [s.Value(step) for s in self.schedules] return tf.reduce_min(tf.stack(ys), axis=0)