Home
last modified time | relevance | path

Searched refs:reduce_to_scalar_loss (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Ddistributed.py9 from torch._dynamo.testing import reduce_to_scalar_loss
28 loss = reduce_to_scalar_loss(outputs)
40 loss = reduce_to_scalar_loss(outputs)
H A Dtimm_models.py18 from torch._dynamo.testing import collect_results, reduce_to_scalar_loss
390 return reduce_to_scalar_loss(pred)
394 return reduce_to_scalar_loss(pred) / 1000.0
H A Ddist_util.py9 from torch._dynamo.testing import reduce_to_scalar_loss
85 loss = reduce_to_scalar_loss(outputs)
H A Dtorchbench.py21 from torch._dynamo.testing import collect_results, reduce_to_scalar_loss
434 return reduce_to_scalar_loss(pred)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtesting.py105 def reduce_to_scalar_loss(out): function
111 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(
H A Ddebug_utils.py316 from .testing import collect_results, reduce_to_scalar_loss, requires_bwd_pass
331 loss = reduce_to_scalar_loss(out)
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_padding.py12 from torch._dynamo.testing import rand_strided, reduce_to_scalar_loss
284 loss = reduce_to_scalar_loss(pred)
H A Dtest_compiled_autograd.py2224 from torch._dynamo.testing import reduce_to_scalar_loss
2229 loss = reduce_to_scalar_loss(out)
2242 from torch._dynamo.testing import reduce_to_scalar_loss
2247 loss = reduce_to_scalar_loss(out)