Searched refs:reduce_to_scalar_loss (Results 1 – 8 of 8) sorted by relevance
9 from torch._dynamo.testing import reduce_to_scalar_loss28 loss = reduce_to_scalar_loss(outputs)40 loss = reduce_to_scalar_loss(outputs)
18 from torch._dynamo.testing import collect_results, reduce_to_scalar_loss390 return reduce_to_scalar_loss(pred)394 return reduce_to_scalar_loss(pred) / 1000.0
9 from torch._dynamo.testing import reduce_to_scalar_loss85 loss = reduce_to_scalar_loss(outputs)
21 from torch._dynamo.testing import collect_results, reduce_to_scalar_loss434 return reduce_to_scalar_loss(pred)
105 def reduce_to_scalar_loss(out): function111 return sum(reduce_to_scalar_loss(x) for x in out) / len(out)117 return reduce_to_scalar_loss(out.logits)121 return sum(reduce_to_scalar_loss(value) for value in out.values()) / len(
316 from .testing import collect_results, reduce_to_scalar_loss, requires_bwd_pass331 loss = reduce_to_scalar_loss(out)
12 from torch._dynamo.testing import rand_strided, reduce_to_scalar_loss284 loss = reduce_to_scalar_loss(pred)
2224 from torch._dynamo.testing import reduce_to_scalar_loss2229 loss = reduce_to_scalar_loss(out)2242 from torch._dynamo.testing import reduce_to_scalar_loss2247 loss = reduce_to_scalar_loss(out)