Searched refs:pfor_input (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 375 def _create_init_values(self, pfor_input): argument
378 loop_len_vector = pfor_input.pfor.loop_len_vector
387 inp, stacked = self._convert_enter(pfor_input.pfor, enter)
434 init_values = [True, pfor_input.pfor.all_indices] + inputs + output_tas
494 def _process_body(self, pfor_input, inputs_stacked, argument
519 body_pfor = self._init_pfor(pfor_input.pfor, new_indices,
541 def __call__(self, pfor_input): argument
585 pfor_input)
610 cond_pfor = self._init_pfor(pfor_input.pfor, indices,
631 new_outputs = self._process_body(pfor_input, inputs_stacked,
[all …]
|