Searched refs:transform_fn (Results 1 – 2 of 2) sorted by relevance
91 transform_fn=None): argument135 if np == 1 and (not transform_fn or ids.get_shape().ndims == 1):139 if transform_fn:140 result = transform_fn(result)209 if transform_fn:213 result = transform_fn(_clip(result, pids, max_norm))220 if transform_fn is None:230 elif transform_fn is None:247 if not transform_fn:328 transform_fn=None)
628 params, ids, max_norm=l2_norm, transform_fn=transform)637 split_params, ids, max_norm=l2_norm, transform_fn=transform)