Searched refs:multi_margin_loss_shape_check (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | LossMulti.h | 37 inline void multi_margin_loss_shape_check( in multi_margin_loss_shape_check() function
|
| D | LossMultiMargin.cpp | 113 multi_margin_loss_shape_check(nframe, dim, ndims, input, target, weight); in multi_margin_loss_out_cpu_template()
225 multi_margin_loss_shape_check(nframe, dim, ndims, input, target, weight); in multi_margin_loss_backward_out_cpu_template()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | MultiMarginLoss.cu | 129 void multi_margin_loss_shape_check( in multi_margin_loss_shape_check() function
172 multi_margin_loss_shape_check(nframe, dim, ndims, input_, target_, weights_); in multi_margin_loss_cuda_out()
308 multi_margin_loss_shape_check(nframe, dim, ndims, input_, target_, weights_); in multi_margin_loss_cuda_backward_out()
|