Searched defs:size_average (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | loss.py | 41 def __init__(self, size_average=None, reduce=None, reduction: str = "mean") -> None: argument
53 size_average=None, argument
124 def __init__(self, size_average=None, reduce=None, reduction: str = "mean") -> None: argument
242 size_average=None, argument
270 size_average=None, argument
343 size_average=None, argument
532 size_average=None, argument
604 def __init__(self, size_average=None, reduce=None, reduction: str = "mean") -> None: argument
690 size_average=None, argument
807 size_average=None, argument
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Loss.cu | 200 bool size_average, in nll_loss_forward_reduce_cuda_kernel_1d()
236 bool size_average, in nll_loss_forward_reduce_cuda_kernel_2d()
435 bool size_average, in nll_loss_backward_reduce_cuda_kernel_1d()
458 bool size_average, in nll_loss_backward_reduce_cuda_kernel_2d()
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | user_defined.py | 249 size_average=ConstantVariable.create(None), argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 11459 def wrapper(input, target, *, size_average=None, reduce=None, reduction="mean", **other_kwargs): argument
|