| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossNLL2d.cpp | 109 int64_t ignore_index) { in nll_loss2d_forward_out_frame()
261 int64_t ignore_index) { in nll_loss2d_forward_out_cpu_template()
290 int64_t ignore_index, in nll_loss2d_backward_out_frame()
373 int64_t ignore_index, in nll_loss2d_backward_out_cpu_template()
410 int64_t ignore_index, in nll_loss2d_forward_out_cpu()
426 int64_t ignore_index) { in nll_loss2d_forward_cpu()
442 int64_t ignore_index, in nll_loss2d_backward_out_cpu()
466 int64_t ignore_index, in nll_loss2d_backward_cpu()
485 …rget, const std::optional<Tensor>& weight_opt, int64_t reduction, int64_t ignore_index, Tensor & o… in nll_loss2d_out()
494 …r & target, const std::optional<Tensor>& weight_opt, int64_t reduction, c10::SymInt ignore_index) { in nll_loss2d_symint()
|
| H A D | LossNLL.cpp | 165 int64_t ignore_index) { in nll_loss_out_frame()
306 int64_t ignore_index) { in nll_loss_forward_out_cpu_template()
344 int64_t ignore_index, in nll_loss_backward_out_frame()
418 int64_t ignore_index, in nll_loss_backward_out_cpu_template()
563 c10::SymInt ignore_index, in cross_entropy_loss_label_smoothing()
629 c10::SymInt ignore_index, in cross_entropy_loss_symint()
661 …rget, const std::optional<Tensor>& weight_opt, int64_t reduction, int64_t ignore_index, Tensor & o… in nll_loss_out()
670 …r & target, const std::optional<Tensor>& weight_opt, int64_t reduction, c10::SymInt ignore_index) { in nll_loss_symint()
683 c10::SymInt ignore_index) { in nll_loss_nd_symint()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Loss.cu | 179 int64_t ignore_index) { in nll_loss_forward_no_reduce_cuda_kernel()
202 int64_t ignore_index) { in nll_loss_forward_reduce_cuda_kernel_1d()
240 int64_t ignore_index) { in nll_loss_forward_reduce_cuda_kernel_2d()
283 int64_t ignore_index) { in nll_loss_forward_out_cuda_template()
415 int64_t ignore_index) { in nll_loss_backward_no_reduce_cuda_kernel()
437 int64_t ignore_index in nll_loss_backward_reduce_cuda_kernel_1d()
462 int64_t ignore_index) { in nll_loss_backward_reduce_cuda_kernel_2d()
489 int64_t ignore_index) { in nll_loss_backward_out_cuda_template()
|
| H A D | NLLLoss2d.cu | 238 int64_t ignore_index) { in nll_loss2d_forward_out_cuda_template()
361 int64_t ignore_index, in nll_loss2d_backward_out_cuda_template()
472 int64_t ignore_index, in nll_loss2d_forward_out_cuda()
485 int64_t ignore_index) { in nll_loss2d_forward_cuda()
499 int64_t ignore_index, in nll_loss2d_backward_out_cuda()
520 int64_t ignore_index, in nll_loss2d_backward_cuda()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset12.py | 116 def nll_loss(g: jit_utils.GraphContext, self, target, weight, reduction, ignore_index): argument
150 g: jit_utils.GraphContext, self, target, weight, reduction, ignore_index argument
157 g: jit_utils.GraphContext, self, target, weight, reduction, ignore_index argument
169 ignore_index, argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_nn.py | 2727 def nlllossNd_reference(input, target, weight=None, ignore_index=-100, argument
2775 def cross_entropy_loss_indices_target_reference(input, target, weight=None, ignore_index=-100, argument
2815 def cross_entropy_loss_reference(input, target, weight=None, ignore_index=-100, reduction='mean', argument
2831 def nllloss_reference(input, target, weight=None, ignore_index=-100, argument
2834 def nll_loss_helper(input, target, weight, ignore_index): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | loss.h | 889 int64_t ignore_index, in nll_loss()
948 int64_t ignore_index, in cross_entropy()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 841 int64_t ignore_index) { in compute_shape_nll_loss2d_forward()
856 int64_t ignore_index, in compute_shape_nll_loss2d_backward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_sharding_util.cc | 552 std::vector<int64_t> ignore_index; in ReshapeToTileDimension() local
558 buckets(ignore_index)[index[dim]].push_back(device); in ReshapeToTileDimension() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | user_defined.py | 250 ignore_index=ConstantVariable.create(-100), argument
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 8951 def _crossentropyloss(self, x, y, ignore_index): argument
8953 def __init__(self, ignore_index): argument
8968 def __init__(self, ignore_index): argument
8987 def __init__(self, ignore_index): argument
9002 def __init__(self, ignore_index): argument
9021 def __init__(self, ignore_index): argument
9034 def __init__(self, ignore_index): argument
|