Home
last modified time | relevance | path

Searched refs:inputs_stacked (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py366 inputs_stacked): argument
388 assert len(inputs_stacked) == len(self._enters)
410 for enter, inp, stacked in zip(self._enters, inputs, inputs_stacked):
414 wrapped_inp = wrap(inputs[i], inputs_stacked[i])
519 inputs_stacked = []
543 inputs_stacked.append(inp_stacked)
554 return init_values, inputs_stacked, shape_invariants
575 def _process_cond_stacked(self, conditions, indices, inputs, inputs_stacked, argument
587 for i, (inp, stacked) in enumerate(zip(inputs, inputs_stacked)):
606 def _process_body(self, pfor_input, inputs_stacked, new_indices, cond_stacked, argument
[all …]