Home
last modified time | relevance | path

Searched refs:binary_cross_entropy_with_logits (Results 1 – 25 of 29) sorted by relevance

12

/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/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/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()
1030 inline Tensor binary_cross_entropy_with_logits(
1034 return detail::binary_cross_entropy_with_logits(
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp282 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()
H A Dautocast_mode.h851 _(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()
/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/torch/_decomp/
H A D__init__.py277 aten.binary_cross_entropy_with_logits,
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dnn_ops.py347 F.binary_cross_entropy_with_logits(torch.sigmoid(a), b),
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_memory_profiler.py1140 loss = torch.nn.functional.binary_cross_entropy_with_logits(y, targets)
1188 loss = torch.nn.functional.binary_cross_entropy_with_logits(y, targets)
H A Dtest_profiler.py1267 loss = torch.nn.functional.binary_cross_entropy_with_logits(z, y)
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/nn/modules/
H A Dloss.py819 return F.binary_cross_entropy_with_logits(
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset12.py211 def binary_cross_entropy_with_logits( function
/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/native/
H A DLoss.cpp362 Tensor binary_cross_entropy_with_logits(const Tensor& input, const Tensor& target, const std::optio… in binary_cross_entropy_with_logits() function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.pyi.in387 def binary_cross_entropy_with_logits(
H A Dfunctional.py3557 def binary_cross_entropy_with_logits( function
3608 binary_cross_entropy_with_logits,
3628 return torch.binary_cross_entropy_with_logits(
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py9342 return torch.nn.functional.binary_cross_entropy_with_logits(
9350 return torch.nn.functional.binary_cross_entropy_with_logits(
9358 return torch.nn.functional.binary_cross_entropy_with_logits(
9367 return torch.nn.functional.binary_cross_entropy_with_logits(
9375 return torch.nn.functional.binary_cross_entropy_with_logits(
9383 return torch.nn.functional.binary_cross_entropy_with_logits(
9392 return torch.nn.functional.binary_cross_entropy_with_logits(
9400 return torch.nn.functional.binary_cross_entropy_with_logits(
9408 return torch.nn.functional.binary_cross_entropy_with_logits(
9417 return torch.nn.functional.binary_cross_entropy_with_logits(
[all …]
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_nn.py264 lambda i: F.binary_cross_entropy_with_logits(i, t.type_as(i), reduce=False)),
282 lambda i: F.binary_cross_entropy_with_logits(i, t.type_as(i), reduction='none')),
300 lambda i: F.binary_cross_entropy_with_logits(i, t.type_as(i), reduction='none')),

12