Searched refs:argnums (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 57 def _grad(f, argnums=0): argument
66 params[argnums],
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 110 def _grad(f, argnums=0): argument
119 params[argnums],
125 def _gradfwd(f, argnums=0, f_out_dtypes=dtypes.float32): argument
132 primals=[params[argnums]], tangents=param_mask) as acc:
139 _single_jvp, [params[argnums]], use_pfor=False, dtype=f_out_dtypes)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_test.py | 1465 def _grad(self, f, argnums=0): argument
1474 params[argnums],
|