Searched refs:transform_fn (Results 1 – 3 of 3) sorted by relevance
89 transform_fn=None): argument131 if np == 1 and (not transform_fn or ids.get_shape().ndims == 1):135 if transform_fn:136 result = transform_fn(result)205 if transform_fn:209 result = transform_fn(_clip(result, pids, max_norm))216 if transform_fn is None:226 elif transform_fn is None:244 if not transform_fn:316 transform_fn=None)
97 weights, sampled, partition_strategy, transform_fn=logsumexp_logit)
628 params, ids, max_norm=l2_norm, transform_fn=transform).eval()638 transform_fn=transform).eval()