Home
last modified time | relevance | path

Searched refs:pfor (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py391 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 …]
Dcontrol_flow_ops_test.py116 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 …]
Dgradients.py63 pfor_outputs = control_flow_ops.pfor(
134 pfor_output = control_flow_ops.pfor(loop_fn, output_row_size,
Dcontrol_flow_ops.py37 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,
Dgradients_test.py179 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__.py23 from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
DBUILD16 "pfor.py",
52 srcs = ["pfor.py"],
Dtest_util.py59 t1 = pfor_control_flow_ops.pfor(
Dxla_control_flow_ops_test.py122 return pfor_control_flow_ops.pfor(loop_fn, 8)
162 return pfor_control_flow_ops.pfor(loop_fn, iters)
Darray_test.py221 pfor_control_flow_ops.pfor(loop_fn, 2, fallback_to_while_loop=False)
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py1175 output = pfor_ops.pfor(loop_fn, target_size,
1311 output = pfor_ops.pfor(loop_fn, target_row_size,
/external/tensorflow/
DRELEASE.md1319 * `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.