Searched defs:maybe_expanded_weight (Results 1 – 1 of 1) sorted by relevance
125 def set_grad_sample_if_exists(maybe_expanded_weight, per_sample_grad_fn): argument149 def unpack_expanded_weight_or_tensor(maybe_expanded_weight, func=lambda x: x): argument