Home
last modified time | relevance | path

Searched refs:binary_cross_entropy (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesLoss.cpp98 return at::binary_cross_entropy(self, target, weight, reduction); in binary_cross_entropy_plumbing()
112 result = at::binary_cross_entropy(self_, target_, std::nullopt, Reduction::None); in binary_cross_entropy_plumbing()
116 result = at::binary_cross_entropy(self_value, target_value, std::nullopt, Reduction::None); in binary_cross_entropy_plumbing()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dloss.h151 inline Tensor binary_cross_entropy( in binary_cross_entropy() function
176 return torch::binary_cross_entropy(input, target, weight_, reduction_enum); in binary_cross_entropy()
195 inline Tensor binary_cross_entropy(
199 return detail::binary_cross_entropy(
/aosp_15_r20/external/pytorch/docs/source/
H A Damp.rst251 Prefer ``binary_cross_entropy_with_logits`` over ``binary_cross_entropy``
253 The backward passes of :func:`torch.nn.functional.binary_cross_entropy` (and :mod:`torch.nn.BCELoss…
257 Therefore, ``binary_cross_entropy`` and ``BCELoss`` raise an error in autocast-enabled regions.
403 ``binary_cross_entropy``,
H A Dnn.functional.rst171 binary_cross_entropy
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp557 auto output = F::binary_cross_entropy(torch::sigmoid(input), target); in TEST_F()
3141 F::binary_cross_entropy(sigmoid(output), target))); in TEST_F()
3149 F::binary_cross_entropy( in TEST_F()
3159 F::binary_cross_entropy(sigmoid(output), target))); in TEST_F()
3167 F::binary_cross_entropy( in TEST_F()
3178 F::binary_cross_entropy( in TEST_F()
H A Dserialize.cpp437 return torch::binary_cross_entropy(x, labels); in TEST()
903 return torch::binary_cross_entropy(x, labels); in TEST()
H A Doptim.cpp54 auto loss = torch::binary_cross_entropy(x, labels); in test_optimizer_xor()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dloss.cpp68 return F::detail::binary_cross_entropy( in forward()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dautocast.cpp493 case aten::binary_cross_entropy: in handleBlock()
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dpublic_api404 binary_cross_entropy
H A Dannotated_ops51 binary_cross_entropy, reduction
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py275 aten.binary_cross_entropy,
H A Ddecompositions.py640 @register_decomposition(aten.binary_cross_entropy)
643 def binary_cross_entropy( function
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dnn_ops.py346 F.binary_cross_entropy(torch.sigmoid(a), b),
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dloss.py697 return F.binary_cross_entropy(
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp354 KERNEL_CPU(binary_cross_entropy, fp32) in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/cpp_api_parity/
H A Dparity-tracker.md225 F::binary_cross_entropy|Yes|No
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.pyi.in379 def binary_cross_entropy(
H A Dfunctional.py3489 def binary_cross_entropy( function
3531 binary_cross_entropy,
3554 return torch._C._nn.binary_cross_entropy(input, target, weight, reduction_enum)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_nn.py193 lambda i: F.binary_cross_entropy(i, t.type_as(i), reduction='none')),
209 lambda i: F.binary_cross_entropy(i, t.type_as(i), reduction='none')),
225 lambda i: F.binary_cross_entropy(i, t.type_as(i),
245 lambda i: F.binary_cross_entropy(i, t.type_as(i),
/aosp_15_r20/external/pytorch/test/
H A Dtest_meta.py362 aten.binary_cross_entropy.default,
712 torch.nn.functional.binary_cross_entropy : {bf16, f64, f32, f16},
H A Dtest_legacy_vmap.py2652 op = functools.partial(F.binary_cross_entropy, target=target)
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_ops.py2315 torch.nn.functional.binary_cross_entropy,
H A Dtest_vmap.py3723 op = functools.partial(F.binary_cross_entropy, target=target)
5230 torch.nn.functional.binary_cross_entropy(escaped, torch.zeros([3, 3, 3, 3]))
/aosp_15_r20/external/pytorch/torch/
H A Doverrides.py845 torch.nn.functional.binary_cross_entropy: (

12