Searched defs:grad_b (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 825 ideep::tensor grad_w, grad_b; in mkldnn_convolution_backward_weights() local
976 ideep::tensor grad_w, grad_b; in mkldnn_convolution_transpose_backward_weights() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | linalg_grad.py | 451 def _LogMatrixDeterminantGrad(op, _, grad_b): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | nn_utils.cpp | 318 torch::Tensor grad_b = torch::ones({10}).mul_(2); in TEST_F() local
|
| H A D | autograd.cpp | 629 const Variable &grad_a = grad_output[0], &grad_b = grad_output[1]; in TEST() local
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_autograd_function.py | 747 def backward(ctx, grad_a, grad_b): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 2519 def backward(ctx, grad_b): argument
3366 def backward(ctx, grad_a, grad_b): argument
3945 def backward(ctx, grad_a, grad_b): argument
3973 def backward(ctx, grad_a, grad_b): argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_aotdispatch.py | 2371 def backward(ctx, grad_a, grad_b): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 4369 Tensor grad_b, grad_a; in triangular_solve_backward() local
|