Searched full:nll_loss2d (Results 1 – 18 of 18) sorted by relevance
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset12.py | 37 "nll_loss2d",
148 @_onnx_symbolic("aten::nll_loss2d")
149 def nll_loss2d( function
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NonSymbolicBC.h | 14 TORCH_API at::Tensor nll_loss2d(const at::Tensor & self, const at::Tensor & target, const std::opti…
|
| D | LossNLL.cpp | 22 #include <ATen/ops/nll_loss2d.h>
|
| D | native_functions.yaml | 11560 - func: nll_loss2d.out(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt …
11563 - func: nll_loss2d(Tensor self, Tensor target, Tensor? weight=None, int reduction=Mean, SymInt igno…
|
| /external/pytorch/aten/src/ATen/ |
| D | autocast_mode.cpp | 270 KERNEL_MPS(nll_loss2d, fp32) in TORCH_LIBRARY_IMPL()
392 KERNEL_CPU(nll_loss2d, fp32) in TORCH_LIBRARY_IMPL()
|
| D | autocast_mode.h | 839 _(nll_loss2d) \
|
| /external/pytorch/functorch/op_analysis/ |
| D | annotated_ops | 456 nll_loss2d, reduction
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesDecompositions.cpp | 219 m.impl("nll_loss2d", native::nll_loss2d_symint); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/torch/testing/_internal/ |
| D | autocast_test_lists.py | 216 ("nll_loss2d", (torch.rand((n, n, n, n), device=dev, dtype=torch.half),
361 ("nll_loss2d", (torch.rand((n, n, n, n), device=dev, dtype=torch.bfloat16),
|
| /external/pytorch/docs/source/ |
| D | amp.rst | 437 ``nll_loss2d``,
|
| D | conf.py | 1374 "nll_loss2d",
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 267 aten::nll_loss2d: 1
|
| D | coverage.yaml | 1068 aten::nll_loss2d: 1
|
| /external/pytorch/torchgen/static_runtime/ |
| D | config.py | 171 if op_name == "nll_loss2d":
|
| D | generator.py | 587 "nll_loss2d",
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | autocast.cpp | 423 case aten::nll_loss2d: in handleBlock()
|
| /external/pytorch/torch/_dynamo/ |
| D | trace_rules.py | 1045 "torch._C._nn.nll_loss2d",
|
| /external/pytorch/test/cpp/lazy/ |
| D | test_lazy_ops.cpp | 8217 torch::Tensor output = torch::nll_loss2d( in TEST_F()
8229 torch::Tensor lazy_output = torch::nll_loss2d( in TEST_F()
11001 return torch::nll_loss2d( in TEST_F()
|