Searched refs:inputs_stacked (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 274 inputs_stacked): argument
296 assert len(inputs_stacked) == len(self._enters)
317 for enter, inp, stacked in zip(self._enters, inputs, inputs_stacked):
321 wrapped_inp = wrap(inputs[i], inputs_stacked[i])
426 inputs_stacked = []
450 inputs_stacked.append(inp_stacked)
461 return init_values, inputs_stacked, shape_invariants
482 def _process_cond_stacked(self, conditions, indices, inputs, inputs_stacked, argument
494 for i, (inp, stacked) in enumerate(zip(inputs, inputs_stacked)):
513 def _process_body(self, pfor_input, inputs_stacked, new_indices, cond_stacked, argument
[all …]
|