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)