Searched defs:expected_gradient (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | xent_op_test_base.py | 63 expected_gradient=None): argument
90 def _testSingleClass(self, expected_gradient=[[2.0], [1.0], [0.0], [0.0]]): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | xent_op_test_base.py | 67 expected_gradient=None): argument
94 def _testSingleClass(self, expected_gradient=[[2.0], [1.0], [0.0], [0.0]]): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer_test.py | 64 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | test_util.py | 29 def create_identity_with_grad_check_fn(expected_gradient, expected_dtype=None): argument
|
| /external/pytorch/test/distributions/ |
| D | test_distributions.py | 5944 expected_gradient=None, argument
|