Home
last modified time | relevance | path

Searched defs:target_contiguous (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLossMultiLabelMargin.cpp63 const Tensor& target_contiguous, in multilabel_margin_loss_forward_out_frame()
147 auto target_contiguous = target.contiguous(); in multilabel_margin_loss_forward_out_cpu_template() local
161 const Tensor& target_contiguous, in multilabel_margin_loss_backward_out_frame()
257 auto target_contiguous = target.contiguous(); in multilabel_margin_loss_backward_out_cpu_template() local
H A DLossMultiMargin.cpp126 auto target_contiguous = target.contiguous(); in multi_margin_loss_out_cpu_template() local
234 auto target_contiguous = target.contiguous(); in multi_margin_loss_backward_out_cpu_template() local
H A DLossNLL2d.cpp175 auto target_contiguous = target.contiguous(); in nll_loss2d_forward_out_frame() local
336 const auto target_contiguous = target.contiguous(); in nll_loss2d_backward_out_frame() local
H A DLossNLL.cpp224 auto target_contiguous = target.contiguous(); in nll_loss_out_frame() local