Searched defs:use_pfor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients.py | 28 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
83 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
|
| D | gradients_test.py | 317 def loop_fn(i, use_pfor): argument
339 def loop_fn(i, use_pfor): argument
|
| D | control_flow_ops_test.py | 1655 def _f(x, y, use_pfor): argument
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 82 def _vectorize_parameters(f, params, use_pfor, dtype): argument
102 def _forward_over_back_hessian(f, params, use_pfor, dtype=None): argument
295 def testHessianOfVariables(self, use_pfor): argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 154 def _vectorize_parameters(f, params, use_pfor, dtype): argument
174 def _forward_over_back_hessian(f, params, use_pfor, dtype=None): argument
|
| D | backprop_test.py | 1854 def compute_jacobian(use_pfor): argument
1970 def test_degenerate_shape(self, use_function, use_pfor): argument
1983 def test_zeros_type_correct(self, use_pfor): argument
|