Lines Matching full:result0

536 …log_probs: _ctc_loss_backward(grad, log_probs, targets, input_lengths, target_lengths, result0, re…
539 …log_probs: _ctc_loss_backward(grad, log_probs, targets, input_lengths, target_lengths, result0, re…
612 …result0: "(!train.has_value() || train.value()) ? (p == 1 ? 0.0 : 1.0 / (1.0 - p)) * input_t * res…
1205 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, running_mean, runnin…
1209 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, running_mean, runnin…
1213 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, running_mean, runnin…
1217 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, Tensor(), Tensor(), …
1226 …result0: layer_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, result1, result2, no…
1236 …result0: group_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, result1, result2, gr…
1255 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, running_mean, runnin…
1259 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, running_mean, runnin…
1623   self: std_mean_backward(grads[0], grads[1], self, result0, dim, correction, keepdim)
1624 …result0: (at::real(var_backward(self_t.conj(), self_p, dim, correction, true).sum(dim.value_or(Int…
1866 …result0: at::real(var_backward(self_t.conj(), self_p, dim, correction, true).sum(dim.value_or(IntA…
2134 …result0: at::where(self_p >= 0, grad_output_t, grad_output_t * weight_p + grad_output_p * weight_t)
2261   result0: gather(self_t.flatten(-2), -1, result1.flatten(-2)).view_as(result1)
2266   result0: gather(self_t.flatten(-3), -1, result1.flatten(-3)).view_as(result1)
2279   result0: gather(self_t.flatten(-2), -1, result1.flatten(-2)).view_as(result1)
2284   result0: gather(self_t.flatten(-3), -1, result1.flatten(-3)).view_as(result1)
2305   result0: gather(self_t.flatten(-2), -1, result1.flatten(-2)).view_as(result1)
2310   result0: gather(self_t.flatten(-3), -1, result1.flatten(-3)).view_as(result1)
2336 …result0: std::get<0>(convolution_backward_symint(grad_output_p, input_p, weight_t, bias_sizes, str…
2650   log_probs: _cudnn_ctc_loss_backward(grad, result0, result1, zero_infinity)
2653   log_probs: _cudnn_ctc_loss_backward(grad, result0, result1, zero_infinity)
2679 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, running_mean, runnin…
2710 …udnn_rnn_backward_symint(input, weight, weight_stride0, result4, hx, cx, result0, grads[0], grads[…
2736 …result0: batch_norm_jvp(input_p, input_t, weight_p, weight_t, bias_p, bias_t, running_mean, runnin…
2746 …ht: "miopen_rnn_backward(input, weight, weight_stride0, result4, hx, cx, result0, grads[0], grads[…
2753 …result0, result1, result2, grads[0], grads[1], grads[2], reverse, mode, hidden_size, num_layers, h…