Searched defs:expected_grad (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_gather_op_test.py | 297 expected_grad, argument
|
| H A D | ragged_tensor_test.py | 1553 def _testGradient(self, func, x, expected_grad, grad_y=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/ |
| H A D | loss_scale_optimizer_test.py | 119 def _run_fn_with_grad_check(self, strategy, var, opt, expected_grad): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
| H A D | gradients_test.cc | 550 const Tensor& expected_grad) { in CheckGrad()
632 const Tensor& expected_grad) { in CheckGrad()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | gradient_checker_test.cc | 36 float* expected_grad, int num_grad, bool use_function, in CompareNumericalAndManualGradients()
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_pooling.py | 1509 def expected_grad(dim, dtype): function
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 862 def _run_and_verify_hook(self, model, input, expected_grad): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | functional.cpp | 3191 const auto expected_grad = torch::empty({3, 1}).fill_(0.5); in TEST_F() local
3274 const auto expected_grad = torch::empty({3, 1}).fill_(0.5); in TEST_F() local
|
| H A D | modules.cpp | 3317 const auto expected_grad = torch::empty({3, 1}).fill_(0.5); in TEST_F() local
3380 const auto expected_grad = torch::empty({3, 1}).fill_(0.5); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 2625 def check(a, q, dim, expected_grad, ops=(torch.quantile, torch.nanquantile)): argument
|