| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLoss.cpp | 98 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 D | loss.h | 151 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 D | amp.rst | 251 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 D | nn.functional.rst | 171 binary_cross_entropy
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | functional.cpp | 557 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 D | serialize.cpp | 437 return torch::binary_cross_entropy(x, labels); in TEST()
903 return torch::binary_cross_entropy(x, labels); in TEST()
|
| H A D | optim.cpp | 54 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 D | loss.cpp | 68 return F::detail::binary_cross_entropy( in forward()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | autocast.cpp | 493 case aten::binary_cross_entropy: in handleBlock()
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 404 binary_cross_entropy
|
| H A D | annotated_ops | 51 binary_cross_entropy, reduction
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 275 aten.binary_cross_entropy,
|
| H A D | decompositions.py | 640 @register_decomposition(aten.binary_cross_entropy)
643 def binary_cross_entropy( function
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | nn_ops.py | 346 F.binary_cross_entropy(torch.sigmoid(a), b),
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | loss.py | 697 return F.binary_cross_entropy(
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 354 KERNEL_CPU(binary_cross_entropy, fp32) in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/cpp_api_parity/ |
| H A D | parity-tracker.md | 225 F::binary_cross_entropy|Yes|No
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.pyi.in | 379 def binary_cross_entropy(
|
| H A D | functional.py | 3489 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 D | common_nn.py | 193 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 D | test_meta.py | 362 aten.binary_cross_entropy.default,
712 torch.nn.functional.binary_cross_entropy : {bf16, f64, f32, f16},
|
| H A D | test_legacy_vmap.py | 2652 op = functools.partial(F.binary_cross_entropy, target=target)
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 2315 torch.nn.functional.binary_cross_entropy,
|
| H A D | test_vmap.py | 3723 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 D | overrides.py | 845 torch.nn.functional.binary_cross_entropy: (
|