/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: (
|