Searched refs:cond_stacked (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 365 def _init_pfor(self, parent_pfor, indices, cond_stacked, inputs, argument
396 all_indices_partitioned=cond_stacked,
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,
723 cond_stacked=True,
726 conditions, cond_stacked, _ = cond_pfor._convert_helper(self._condition)
727 cond_is_stacked[0] = cond_stacked
731 if not cond_stacked:
745 new_indices, cond_stacked, new_inputs,
3277 cond, cond_stacked, _ = pfor_input.input(0)
[all …]
|