Searched refs:pfor_input (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 488 def _create_init_values(self, pfor_input): argument
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
606 def _process_body(self, pfor_input, inputs_stacked, new_indices, cond_stacked, argument
630 body_pfor = self._init_pfor(pfor_input.pfor, new_indices, cond_stacked,
651 def __call__(self, pfor_input): argument
695 pfor_input)
721 pfor_input.pfor,
744 new_outputs = self._process_body(pfor_input, inputs_stacked,
[all …]
|