Home
last modified time | relevance | path

Searched full:binary_cross_entropy_with_logits (Results 1 – 25 of 48) sorted by relevance

12

/aosp_15_r20/external/pytorch/docs/source/
H A Damp.rst182 ``binary_cross_entropy_with_logits``,
251 Prefer ``binary_cross_entropy_with_logits`` over ``binary_cross_entropy``
260 …his case, combine the two layers using :func:`torch.nn.functional.binary_cross_entropy_with_logits`
261 or :mod:`torch.nn.BCEWithLogitsLoss`. ``binary_cross_entropy_with_logits`` and ``BCEWithLogits``
313 ``binary_cross_entropy_with_logits``,
450 ``binary_cross_entropy_with_logits``,
H A Dnn.functional.rst172 binary_cross_entropy_with_logits
/aosp_15_r20/external/pytorch/torch/distributions/
H A Dbernoulli.py14 from torch.nn.functional import binary_cross_entropy_with_logits
113 return -binary_cross_entropy_with_logits(logits, value, reduction="none")
116 return binary_cross_entropy_with_logits(
H A Dgeometric.py13 from torch.nn.functional import binary_cross_entropy_with_logits
128 binary_cross_entropy_with_logits(self.logits, self.probs, reduction="none")
H A Dcontinuous_bernoulli.py15 from torch.nn.functional import binary_cross_entropy_with_logits
181 -binary_cross_entropy_with_logits(logits, value, reduction="none")
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset12.py28 "binary_cross_entropy_with_logits",
209 @_onnx_symbolic("aten::binary_cross_entropy_with_logits")
211 def binary_cross_entropy_with_logits( function
249 "binary_cross_entropy_with_logits with reduction other than none, mean, or sum",
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp154 …"In this case, combine the two layers using torch.nn.functional.binary_cross_entropy_with_logits\n" in binary_cross_entropy_banned()
155 … "or torch.nn.BCEWithLogitsLoss. binary_cross_entropy_with_logits and BCEWithLogits are\n" in binary_cross_entropy_banned()
282 KERNEL_MPS(binary_cross_entropy_with_logits, fp32) in TORCH_LIBRARY_IMPL()
407 KERNEL_CPU(binary_cross_entropy_with_logits, fp32) in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp3122 F::binary_cross_entropy_with_logits(input, target), in TEST_F()
3129 F::binary_cross_entropy_with_logits(input, target), in TEST_F()
3140 F::binary_cross_entropy_with_logits(output, target), in TEST_F()
3145 F::binary_cross_entropy_with_logits( in TEST_F()
3158 F::binary_cross_entropy_with_logits(output, target), in TEST_F()
3162 F::binary_cross_entropy_with_logits( in TEST_F()
3174 F::binary_cross_entropy_with_logits( in TEST_F()
3186 F::binary_cross_entropy_with_logits( in TEST_F()
3199 auto out1 = F::binary_cross_entropy_with_logits( in TEST_F()
3205 auto out2 = F::binary_cross_entropy_with_logits( in TEST_F()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dloss.h992 inline Tensor binary_cross_entropy_with_logits( in binary_cross_entropy_with_logits() function
1006 return torch::binary_cross_entropy_with_logits( in binary_cross_entropy_with_logits()
1017 …tps://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.binary_cross_entropy_with_logits
1027 /// F::binary_cross_entropy_with_logits(input, target,
1030 inline Tensor binary_cross_entropy_with_logits(
1034 return detail::binary_cross_entropy_with_logits(
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dtop_operators_github_usage.py292 ("nn.functional.binary_cross_entropy_with_logits", 591),
427 ("nn.BCEWithLogitsLoss", 29944, "nn.functional.binary_cross_entropy_with_logits"),
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_memory_profiler.py1140 loss = torch.nn.functional.binary_cross_entropy_with_logits(y, targets)
1155 …aten::binary_cross_entropy_with_logits 7 (INPUT), 2 (INPUT) -> 11 (INPU…
1188 loss = torch.nn.functional.binary_cross_entropy_with_logits(y, targets)
1207 …aten::binary_cross_entropy_with_logits 7 (ACTIVATION), 2 (INPUT) -> 11 (ACTI…
H A Dtest_profiler.py1267 loss = torch.nn.functional.binary_cross_entropy_with_logits(z, y)
1308 ts_to_name[s_ts_1] == "aten::binary_cross_entropy_with_logits"
1322 loss = torch.nn.functional.binary_cross_entropy_with_logits(z, y)
2224 loss = torch.nn.functional.binary_cross_entropy_with_logits(z, y)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dautocast_test_lists.py166 …("binary_cross_entropy_with_logits", mat0_fp16 + (torch.rand((n, n), device=dev, dtype=torch.float…
349 …("binary_cross_entropy_with_logits", mat0_bf16 + (torch.rand((n, n), device=dev, dtype=torch.bfloa…
H A Dcommon_nn.py264 lambda i: F.binary_cross_entropy_with_logits(i, t.type_as(i), reduce=False)),
265 cpp_function_call='''F::binary_cross_entropy_with_logits(
282 lambda i: F.binary_cross_entropy_with_logits(i, t.type_as(i), reduction='none')),
283 cpp_function_call='''F::binary_cross_entropy_with_logits(
300 lambda i: F.binary_cross_entropy_with_logits(i, t.type_as(i), reduction='none')),
301 cpp_function_call='''F::binary_cross_entropy_with_logits(
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dpublic_api405 binary_cross_entropy_with_logits
H A Dannotated_ops52 binary_cross_entropy_with_logits, reduction
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml108 - aten::binary_cross_entropy_with_logits
761 aten::binary_cross_entropy_with_logits: 3
H A Dmodel_ops.yaml82 aten::binary_cross_entropy_with_logits: 6
H A Dnn_ops.py347 F.binary_cross_entropy_with_logits(torch.sigmoid(a), b),
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dloss.h787 /// Options for `torch::nn::functional::binary_cross_entropy_with_logits`.
795 /// F::binary_cross_entropy_with_logits(input, target,
/aosp_15_r20/external/pytorch/test/cpp_api_parity/
H A Dparity-tracker.md226 F::binary_cross_entropy_with_logits|Yes|No
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DPyTorchOperatorHacks.cpp261 m.impl("binary_cross_entropy_with_logits", binary_cross_entropy_with_logits_hack); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py277 aten.binary_cross_entropy_with_logits,
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dloss.cpp428 return F::detail::binary_cross_entropy_with_logits( in forward()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dautocast.cpp434 case aten::binary_cross_entropy_with_logits: in handleBlock()

12