Searched defs:reinterpreted_batch_ndims (Results 1 – 3 of 3) sorted by relevance
46 self, base_distribution, reinterpreted_batch_ndims, validate_args=None argument
227 def __init__(self, base_constraint, reinterpreted_batch_ndims): argument
416 def __init__(self, base_transform, reinterpreted_batch_ndims, cache_size=0): argument