Searched refs:multilabel_margin_loss_shape_check (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | LossMulti.h | 8 inline void multilabel_margin_loss_shape_check( in multilabel_margin_loss_shape_check() function
|
| D | LossMultiLabelMargin.cpp | 123 multilabel_margin_loss_shape_check(nframe, dim, ndims, input, target); in multilabel_margin_loss_forward_out_cpu_template()
240 multilabel_margin_loss_shape_check(nframe, dim, ndims, input, target); in multilabel_margin_loss_backward_out_cpu_template()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | MultiLabelMarginCriterion.cu | 26 void multilabel_margin_loss_shape_check( in multilabel_margin_loss_shape_check() function
208 multilabel_margin_loss_shape_check(nframe, dim, ndims, input, target); in multilabel_margin_loss_forward_out_cuda_template()
313 multilabel_margin_loss_shape_check(nframe, dim, ndims, input, target); in multilabel_margin_loss_backward_cuda_out_template()
|