Searched refs:body_pfor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 610 def true_fn(control_inputs, body_pfor, body_output, stacked): argument
620 converted_control_inp.append(body_pfor._convert_helper(t).t)
623 output = body_pfor.convert(body_output)
625 output, convert_stacked, _ = body_pfor._convert_helper(body_output)
630 body_pfor = self._init_pfor(pfor_input.pfor, new_indices, cond_stacked,
645 lambda: true_fn(control_inp, body_pfor, body_output, stacked),
4421 body_pfor = PFor(
4432 body_pfor,
|