Searched defs:target_arg (Results 1 – 3 of 3) sorted by relevance
118 auto target_arg = TensorArg(target, "target", 2); in multilabel_margin_loss_forward_out_cpu_template() local236 auto target_arg = TensorArg(target, "target", 3); in multilabel_margin_loss_backward_out_cpu_template() local
1003 def _get_arg_indices(self, target_arg): argument
1281 def _get_arg_indices(self, target_arg): argument