Searched defs:use_pfor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients.py | 24 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
79 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
|
| D | gradients_test.py | 313 def loop_fn(i, use_pfor): argument
335 def loop_fn(i, use_pfor): argument
|
| D | control_flow_ops_test.py | 2015 def _f(x, y, use_pfor): argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 155 def _vectorize_parameters(f, params, use_pfor, dtype): argument
175 def _forward_over_back_hessian(f, params, use_pfor, dtype=None): argument
|
| D | backprop_test.py | 1911 def compute_jacobian(use_pfor): argument
2027 def test_degenerate_shape(self, use_function, use_pfor): argument
2040 def test_zeros_type_correct(self, use_pfor): argument
|