def clip(values_blob):
with flow.scope.placement(device_type, "0:0"):
x = flow.get_variable(
"values",
shape=values.shape,
dtype=data_type,
initializer=flow.constant_initializer(0),
)
x = flow.cast_to_current_logical_view(x)
x = x + values_blob
y = flow.clip_by_value(x, min, max)
flow.losses.add_loss(y)
flow.watch_diff(x, grad_cb)
return y
def clip(values_blob):
with flow.scope.placement(device_type, "0:0"):
x = flow.get_variable(
"values",
shape=values.shape,
dtype=data_type,
initializer=flow.constant_initializer(0),
)
x = flow.cast_to_current_logical_view(x)
x = x + values_blob
y = flow.clip_by_value(x, min, max)
flow.optimizer.SGD(
flow.optimizer.PiecewiseConstantScheduler([], [1e-3]), momentum=0
).minimize(y)
flow.watch_diff(x, grad_cb)
return y
def clip(values_blob):
with flow.scope.placement(device_type, "0:0"):
return flow.clip_by_value(values_blob, min, max, name="Clip")
def _relu6(data, prefix):
return flow.clip_by_value(data, 0, 6, name="%s-relu6" % prefix)
def _relu6(data):
return flow.clip_by_value(data, 0, 6)
