Searched defs:weight_contiguous (Results 1 – 3 of 3) sorted by relevance
127 Tensor weight_contiguous; in multi_margin_loss_out_cpu_template() local235 auto weight_contiguous = weight.contiguous(); in multi_margin_loss_backward_out_cpu_template() local
115 auto weight_contiguous = optional_contiguous(weight); in nll_loss2d_forward_out_frame() local292 auto weight_contiguous = optional_contiguous(weight); in nll_loss2d_backward_out_frame() local
172 auto weight_contiguous = optional_contiguous(weight); in nll_loss_out_frame() local355 auto weight_contiguous = optional_contiguous(weight); in nll_loss_backward_out_frame() local