Searched refs:pfor (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 391 pfor = PFor(
407 pfor._add_conversion(enter, wrap(converted_enter, False))
411 pfor._add_conversion(enter, wrap(inp, stacked))
416 pfor._add_conversion(merge.outputs[0], wrapped_inp)
425 pfor._add_conversion(merge.outputs[1],
429 pfor._add_conversion(switch.outputs[1], wrapped_inp)
430 return pfor
491 loop_len_vector = pfor_input.pfor.loop_len_vector
500 inp, stacked = self._convert_enter(pfor_input.pfor, enter)
547 init_values = [True, pfor_input.pfor.all_indices] + inputs + output_tas
[all …]
|
| D | control_flow_ops_test.py | 116 x = pfor_control_flow_ops.pfor(
124 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=0)
131 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=1)
234 output = pfor_control_flow_ops.pfor(loop_fn, 4)
248 output = pfor_control_flow_ops.pfor(loop_fn, 8)
260 output = pfor_control_flow_ops.pfor(loop_fn, 8)
272 output = pfor_control_flow_ops.pfor(loop_fn, 8)
289 output = pfor_control_flow_ops.pfor(LoopFn(), 8)
303 output = pfor_control_flow_ops.pfor(loop_fn, 8)
317 pfor_control_flow_ops.pfor(loop_fn, 8, parallel_iterations=2)
[all …]
|
| D | gradients.py | 63 pfor_outputs = control_flow_ops.pfor(
134 pfor_output = control_flow_ops.pfor(loop_fn, output_row_size,
|
| D | control_flow_ops.py | 37 from tensorflow.python.ops.parallel_for.pfor import PFor
38 from tensorflow.python.ops.parallel_for.pfor import PForConfig
138 def pfor(loop_fn, iters, fallback_to_while_loop=True, parallel_iterations=None): function
546 return pfor(loop_fn, batch_size,
|
| D | gradients_test.py | 179 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
208 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
273 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
296 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
323 pfor_outputs = control_flow_ops.pfor(
345 per_eg_jacobians_pfor = control_flow_ops.pfor(
504 pfor = self.evaluate(pfor_jacobian)
507 self.assertAllClose(while_i, pfor[:, i, ...])
|
| D | __init__.py | 23 from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
|
| D | BUILD | 16 "pfor.py",
52 srcs = ["pfor.py"],
|
| D | test_util.py | 59 t1 = pfor_control_flow_ops.pfor(
|
| D | xla_control_flow_ops_test.py | 122 return pfor_control_flow_ops.pfor(loop_fn, 8)
162 return pfor_control_flow_ops.pfor(loop_fn, iters)
|
| D | array_test.py | 221 pfor_control_flow_ops.pfor(loop_fn, 2, fallback_to_while_loop=False)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1175 output = pfor_ops.pfor(loop_fn, target_size,
1311 output = pfor_ops.pfor(loop_fn, target_row_size,
|
| /external/tensorflow/ |
| D | RELEASE.md | 1319 * `pfor/vectorized_map`: Added support for vectorizing 56 more ops.
1611 * Add `pfor` converter for `SelfAdjointEigV2`.
2095 * `parallel_for.pfor`: add converters for Softmax, LogSoftmax, IsNaN, All,
2103 * Add `pfor` converter for `Squeeze`.
2424 * Add pfor converter for Squeeze.
|