def FProp(self, theta, current_step):
p = self.params
return py_utils.PiecewiseConstant(current_step, p.boundaries, p.values,
p.dtype)
def Value(self):
p = self.params
return py_utils.PiecewiseConstant(py_utils.GetGlobalStep(), p.boundaries,
p.values, p.dtype)
def _Eval(x):
with self.session(use_gpu=False) as sess:
result = py_utils.PiecewiseConstant(
x, boundaries, values, vdtype=tf.float32)
return sess.run(result)