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)
