Home
last modified time | relevance | path

Searched full:binary_cross_entropy (Results 1 – 25 of 52) sorted by relevance

123

/aosp_15_r20/external/pytorch/test/cpp_api_parity/
H A Dutils.py61 # NN functional name (e.g. "binary_cross_entropy")
69 # (e.g. "F::binary_cross_entropy(
71 …# which is strictly equivalent to `F.binary_cross_entropy(i, t.type_as(i), reduction='none')` in P…
284 # lambda i: F.binary_cross_entropy(i, t.type_as(i),
286 # cpp_function_call='''F::binary_cross_entropy(
H A Dfunctional_impl_check.py148 # Example output: `binary_cross_entropy`
157 # Example output: `binary_cross_entropy`
H A Dparity-tracker.md225 F::binary_cross_entropy|Yes|No
/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()
179 m.impl("binary_cross_entropy", binary_cross_entropy_plumbing); in TORCH_LIBRARY_IMPL()
/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/aten/src/ATen/
H A Dautocast_mode.cpp152 AT_ERROR("torch.nn.functional.binary_cross_entropy and torch.nn.BCELoss are unsafe to autocast.\n" in binary_cross_entropy_banned()
205 m.impl(TORCH_SELECTIVE_NAME("aten::binary_cross_entropy"), in TORCH_LIBRARY_IMPL()
354 KERNEL_CPU(binary_cross_entropy, fp32) in TORCH_LIBRARY_IMPL()
496 m.impl(TORCH_SELECTIVE_NAME("aten::binary_cross_entropy"), in TORCH_LIBRARY_IMPL()
/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()
182 /// https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.binary_cross_entropy
192 /// F::binary_cross_entropy(input, target,
195 inline Tensor binary_cross_entropy(
199 return detail::binary_cross_entropy(
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dtop_operators_github_usage.py294 ("nn.functional.binary_cross_entropy", 492),
425 ("nn.BCELoss", 34005, "nn.functional.binary_cross_entropy"),
/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')),
194 cpp_function_call='F::binary_cross_entropy('
209 lambda i: F.binary_cross_entropy(i, t.type_as(i), reduction='none')),
210 cpp_function_call='F::binary_cross_entropy('
225 lambda i: F.binary_cross_entropy(i, t.type_as(i),
227 cpp_function_call='F::binary_cross_entropy('
245 lambda i: F.binary_cross_entropy(i, t.type_as(i),
247 cpp_function_call='''F::binary_cross_entropy(
H A Dautocast_test_lists.py236 ("binary_cross_entropy", (torch.rand((n, n), device=dev, dtype=torch.float32),
356 ("binary_cross_entropy", (torch.rand((n, n), device=dev, dtype=torch.bfloat16),) +
H A Djit_metaprogramming_utils.py250 ('binary_cross_entropy', torch.randn(3, 2).sigmoid(), (non_differentiable(torch.rand(3, 2)),
252 ('binary_cross_entropy', torch.randn(3, 2).sigmoid(),
H A Dcommon_jit.py37 'binary_cross_entropy',
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml23 - binary_cross_entropy
/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/test/mobile/model_test/
H A Dcoverage.yaml107 - aten::binary_cross_entropy
760 aten::binary_cross_entropy: 15
H A Dmodel_ops.yaml81 aten::binary_cross_entropy: 16
H A Dnn_ops.py346 F.binary_cross_entropy(torch.sigmoid(a), b),
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dloss.h142 /// Options for `torch::nn::functional::binary_cross_entropy`.
150 /// F::binary_cross_entropy(input, target,
/aosp_15_r20/external/pytorch/test/cpp/api/
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 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()
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py275 aten.binary_cross_entropy,
/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()

123