Searched defs:expanded_weight (Results 1 – 2 of 2) sorted by relevance
118 def maybe_scale_by_batch_size(grad_sample, expanded_weight): argument
147 Value* expanded_weight = in DecomposeOps() local