Searched refs:use_pfor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 106 pfor_jacobian = gradients.batch_jacobian(output, inp, use_pfor=True)
107 while_jacobian = gradients.batch_jacobian(output, inp, use_pfor=False)
114 pfor_jacobian = gradients.batch_jacobian(output, inp, use_pfor=True)
115 while_jacobian = gradients.batch_jacobian(output, inp, use_pfor=False)
122 pfor_jacobian = gradients.batch_jacobian(final_state.c, inp, use_pfor=True)
135 pfor_jacobian = gradients.batch_jacobian(output, inp, use_pfor=True)
137 pfor_hessian = gradients.batch_jacobian(pfor_jacobian, inp, use_pfor=True)
141 while_hessian = gradients.batch_jacobian(while_jacobian, inp, use_pfor=False)
148 pfor_jacobians = gradients.jacobian(output, weights, use_pfor=True)
150 gradients.jacobian(x, weights, use_pfor=True) for x in pfor_jacobians
[all …]
|
| D | gradients.py | 28 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
62 if use_pfor:
83 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
133 if use_pfor:
|
| D | control_flow_ops_test.py | 1655 def _f(x, y, use_pfor): argument
1666 if use_pfor:
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 82 def _vectorize_parameters(f, params, use_pfor, dtype): argument
96 if use_pfor:
102 def _forward_over_back_hessian(f, params, use_pfor, dtype=None): argument
122 use_pfor=use_pfor,
295 def testHessianOfVariables(self, use_pfor): argument
306 use_pfor=use_pfor,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 139 _single_jvp, [params[argnums]], use_pfor=False, dtype=f_out_dtypes)
154 def _vectorize_parameters(f, params, use_pfor, dtype): argument
168 if use_pfor:
174 def _forward_over_back_hessian(f, params, use_pfor, dtype=None): argument
194 use_pfor=use_pfor,
1014 use_pfor=False,
1019 use_pfor=False,
1024 use_pfor=True,
|
| D | backprop_test.py | 1854 def compute_jacobian(use_pfor): argument
1860 return tape.jacobian(result, x, experimental_use_pfor=use_pfor)
1862 self.assertAllClose(compute_jacobian(use_pfor=True),
1863 compute_jacobian(use_pfor=False))
1970 def test_degenerate_shape(self, use_function, use_pfor): argument
1976 return tape.batch_jacobian(y, x, experimental_use_pfor=use_pfor)
1983 def test_zeros_type_correct(self, use_pfor): argument
1994 jac = tape.batch_jacobian(y, x, experimental_use_pfor=use_pfor)
2003 experimental_use_pfor=use_pfor)
|