Searched refs:pfor (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 398 pfor = PFor(
414 pfor._add_conversion(enter, wrap(converted_enter, False))
418 pfor._add_conversion(enter, wrap(inp, stacked))
423 pfor._add_conversion(merge.outputs[0], wrapped_inp)
432 pfor._add_conversion(merge.outputs[1],
436 pfor._add_conversion(switch.outputs[1], wrapped_inp)
437 return pfor
498 loop_len_vector = pfor_input.pfor.loop_len_vector
507 inp, stacked = self._convert_enter(pfor_input.pfor, enter)
554 init_values = [True, pfor_input.pfor.all_indices] + inputs + output_tas
[all …]
|
| D | control_flow_ops_test.py | 118 x = pfor_control_flow_ops.pfor(
126 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=0)
133 pfor_control_flow_ops.pfor(lambda i: 1, 8, parallel_iterations=1)
247 output = pfor_control_flow_ops.pfor(loop_fn, 4)
261 output = pfor_control_flow_ops.pfor(loop_fn, 8)
273 output = pfor_control_flow_ops.pfor(loop_fn, 8)
285 output = pfor_control_flow_ops.pfor(loop_fn, 8)
302 output = pfor_control_flow_ops.pfor(LoopFn(), 8)
316 output = pfor_control_flow_ops.pfor(loop_fn, 8)
330 pfor_control_flow_ops.pfor(loop_fn, 8, parallel_iterations=2)
[all …]
|
| D | index.md | 49 This prints a bunch of gathers related to pfor infrastructure, but at the time
107 …low/blob/349172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L2653)
112 vectorizable). In practice not all ops have pfor converters defined, and those
120 common case, pfor converters take loop-variant inputs with an extra dimension
135 tensorflow/python/ops/parallel_for/pfor.py (the user-facing APIs are defined in
152 …9172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L3505-L3522) ([e…
154 …9172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L3276-L3294) sha…
156 …ob/349172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L3360-L3377)
171 …9172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L4499); [example…
179 …9172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L5001) is fairly…
[all …]
|
| D | gradients.py | 59 pfor_outputs = control_flow_ops.pfor(
130 pfor_output = control_flow_ops.pfor(loop_fn, output_row_size,
|
| D | control_flow_ops.py | 35 from tensorflow.python.ops.parallel_for.pfor import PFor
36 from tensorflow.python.ops.parallel_for.pfor import PForConfig
138 def pfor(loop_fn, function
565 return pfor(
|
| D | gradients_test.py | 175 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
204 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
269 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
292 pfor_outputs = control_flow_ops.pfor(loop_fn, batch_size)
319 pfor_outputs = control_flow_ops.pfor(
341 per_eg_jacobians_pfor = control_flow_ops.pfor(
500 pfor = self.evaluate(pfor_jacobian)
503 self.assertAllClose(while_i, pfor[:, i, ...])
|
| D | __init__.py | 19 from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
|
| D | BUILD | 16 "pfor.py",
53 srcs = ["pfor.py"],
|
| D | test_util.py | 55 t1 = pfor_control_flow_ops.pfor(
|
| D | xla_control_flow_ops_test.py | 118 return pfor_control_flow_ops.pfor(loop_fn, 8)
158 return pfor_control_flow_ops.pfor(loop_fn, iters)
|
| D | array_test.py | 236 pfor_control_flow_ops.pfor(loop_fn, 2, fallback_to_while_loop=False)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1216 output = pfor_ops.pfor(loop_fn, target_size,
1363 output = pfor_ops.pfor(loop_fn, target_row_size,
|
| /external/tensorflow/ |
| D | RELEASE.md | 5823 * `pfor/vectorized_map`: Added support for vectorizing 56 more ops.
6223 * Add `pfor` converter for `SelfAdjointEigV2`.
6954 * `parallel_for.pfor`: add converters for Softmax, LogSoftmax, IsNaN, All,
6962 * Add `pfor` converter for `Squeeze`.
7283 * Add pfor converter for Squeeze.
|